Spaces:
Runtime error
Runtime error
Update mnist_model.pth
Browse files- mnist_model.pth +2 -0
mnist_model.pth
CHANGED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Save the trained model state dictionary
|
| 2 |
+
torch.save(model.state_dict(), 'mnist_model.pth')
|