Spaces:

danielcd99
/

MNIST-classifier

Runtime error

danielcd99 commited on Apr 15, 2023

Commit

fca1bbb

•

1 Parent(s): 16e1dde

Commented train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -86,4 +87,6 @@ def train(model):
             loss.backward()
             optimizer.step()
-torch.save(model_adam.state_dict(), "model_mnist.pth")

+"""
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
             loss.backward()
             optimizer.step()
+# SAVING MODEL
+torch.save(model_adam.state_dict(), "model_mnist.pth")
+"""