Upload model

modeling_spice_cnn.py

@@ -20,13 +20,8 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             nn.BatchNorm2d(32),
             nn.ReLU(),
             nn.MaxPool2d(kernel_size=config.pooling_size),
-            
-            nn.Conv2d(32, 64, kernel_size=config.kernel_size, padding=1),
-            nn.BatchNorm2d(64),
-            nn.ReLU(),
-            nn.MaxPool2d(kernel_size=config.pooling_size),
 
-            nn.Linear(64*2*2, 128),
+            nn.Linear(7*7*32, 128),
             nn.ReLU(),
             nn.Dropout(0.5),
             nn.Linear(128, config.num_classes)
@@ -39,4 +34,4 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
             loss_fnc = nn.CrossEntropyLoss()
             loss = loss_fnc(logits, labels)
             return {"loss": loss, "logits": logits}
-        return {"logits": logits}
+        return {"logits": logits}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:27bc8f3fbf442d90430939e9d7d9569b01ddb1f98c7fd9a026784d6f01471b55
-size 238972
+oid sha256:1d40410140194428e35927afeb15389e18b894f35bd109991e281597b3938623
+size 833767