Upload model

config.json

@@ -8,6 +8,7 @@
   },
   "dropout_rate": 0.2,
   "hidden_size": 128,
+  "in_channels": 1,
   "kernel_size": 3,
   "model_type": "spicecnn",
   "num_classes": 10,

configuration_spice_cnn.py

@@ -26,6 +26,7 @@ class SpiceCNNConfig(PretrainedConfig):
 
     def __init__(
         self,
+        in_channels: int = 3,
         num_classes: int = 10,
         dropout_rate: float = 0.2,
         hidden_size: int = 128,
@@ -37,6 +38,7 @@ class SpiceCNNConfig(PretrainedConfig):
         **kwargs
     ):
         super().__init__(**kwargs)
+        self.in_channels = in_channels
         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(
-                1,
+                config.in_channels,
                 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:e07e343bdf0c9f3c218ba10e865eca0d59c3adf4da998294d288160c382657e6
+oid sha256:b2bd33ae4006b549f8ef4839e107525981d190d4922a7236e5de3a59190450a1
 size 830347