model architecture

Author: Deep1994

lenet5/lenet5.py

@@ -36,4 +36,8 @@ def forward(self, x):
         out = self.convnet(x)
         out = out.view(x.size(0), -1)
         out = self.fc(out)
-        return out
+<<<<<<< HEAD
+        return out
+=======
+        return out
+>>>>>>> fb8de6a3d021d0f15290d23e9c70cc89f5564c4b

lenet5/main.py

@@ -98,4 +98,8 @@
     print('Test Accuracy of the model on the 10000 test images: {} %'.format(100 * correct / total))
 
 # Save the model checkpoint
-torch.save(model.state_dict(), 'model.ckpt')
+<<<<<<< HEAD
+torch.save(model.state_dict(), 'model.ckpt')
+=======
+torch.save(model.state_dict(), 'model.ckpt')
+>>>>>>> fb8de6a3d021d0f15290d23e9c70cc89f5564c4b