Upload model

modeling_spice_cnn.py

@@ -37,10 +37,10 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.Dropout(0.4),
             
             nn.Flatten(),
-            nn.BatchNorm1d(128),
+            nn.BatchNorm1d(64),
             nn.ReLU(),
             nn.Dropout(0.4),
-            nn.Linear(128, config.num_classes),
+            nn.Linear(64, config.num_classes),
         ]
         self.model = nn.Sequential(*layers)
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:13d74c7b24138f2a238170dceb0626116f1c7a572d91ce364530e8945482b2c8
-size 796341
+oid sha256:46f937126b04976c9954c667a252be6c1ae0e7283a81c620e8ed53d50d7dcded
+size 792757