Upload model

modeling_spice_cnn.py

@@ -37,7 +37,7 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.Dropout(0.4),
             
             nn.Flatten(),
-            nn.Linear(64 * 5 * 5, 128),
+            nn.Linear(14*14*5, 128),
             nn.BatchNorm1d(128),
             nn.ReLU(),
             nn.Dropout(0.4),

pytorch_model.bin

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