Upload model

configuration_spice_cnn.py

@@ -37,7 +37,6 @@ class SpiceCNNConfig(PretrainedConfig):
         **kwargs
     ):
         super().__init__(**kwargs)
-
         self.num_classes = num_classes
         self.dropout_rate = dropout_rate
         self.hidden_size = hidden_size

modeling_spice_cnn.py

@@ -12,7 +12,7 @@ class SpiceCNNModelForImageClassification(PreTrainedModel):
         super().__init__(config)
         layers = [
             nn.Conv2d(
-                3,
+                1,
                 16,
                 kernel_size=config.kernel_size,
                 stride=config.stride,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ccf52d67db298de17ef9b79093625ffa2c10ca80eb0c5a8337414b30d6e2ca7f
-size 831499
+oid sha256:e192c133a897fdd08ebce05f5c8f022ee987046194f4c4a97d7ece5acb9a854b
+size 830347