Upload model

modeling_spice_cnn.py

@@ -12,12 +12,11 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
         super().__init__(config)
         layers = [
             nn.Conv2d(
-                1,
+                3,
                 32,
                 kernel_size=config.kernel_size,
                 stride=config.stride,
                 padding=config.padding,
-                groups=1,
             ),
             nn.ReLU(),
             nn.MaxPool2d(kernel_size=config.pooling_size),
@@ -27,7 +26,6 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
                 kernel_size=config.kernel_size,
                 stride=config.stride,
                 padding=config.padding,
-                groups=32,
             ),
             nn.ReLU(),
             nn.MaxPool2d(kernel_size=config.pooling_size),

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37521dcfc2224b7bc741e21c5852a652e2076569070428c8657e3b95ca86684e
-size 1617803
+oid sha256:a79aabbf2241069a38ecec45da166aba5195788cdeafa15a147c09fce443ca2b
+size 1691531