10215501417
/
mnist
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
Browse Source

commit source code and model

master
10215501417 1 year ago
parent
d80c3280fd
commit
884cd786dd
3 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      code/file_assignment7_mnist.py
  2. +0
    -0
      output/model_epoch_10

+ 1
- 1
code/file_assignment7_mnist.py View File

@ -55,4 +55,4 @@ plt.savefig('./mnist/output/2.jpg')
# np.argmax(model(x_test[:9]).numpy(), axis=1)
#保存训练好的模型
model.save("./mnist/output/model_epoch_5")
model.save("./mnist/output/model_epoch_10")

output/model_epoch_5 → output/model_epoch_10 View File


Write Preview
Loading…
Cancel
Save