Upload model

modeling_spice_cnn.py

@@ -37,8 +37,9 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.Dropout(0.4),
             
             nn.Flatten(),
-            nn.Linear(64 * 28 * 28, 128),
+            nn.Linear(64 * 5 * 5, 128),
             nn.BatchNorm1d(128),
+            nn.ReLU(),
             nn.Dropout(0.4),
             nn.Linear(128, config.num_classes),
         ]

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:60d140c12a803582c4e3e7b0d2e52fbfb492477523541fa0ec719ce40b47358d
-size 26487547
+oid sha256:60777acb6fdfb226898e754740abf81831deff984d549ee2645db4aa7fa16c29
+size 1616635