config.json

{
  "architectures": [
    "SpiceCNNModelForImageClassification"
  ],
  "auto_map": {
    "AutoConfig": "configuration_spice_cnn.SpiceCNNConfig",
    "AutoModelForImageClassification": "modeling_spice_cnn.SpiceCNNModelForImageClassification"
  },
  "dropout_rate": 0.4,
  "hidden_size": 128,
  "in_channels": 1,
  "kernel_size": 3,
  "model_type": "spicecnn",
  "num_classes": 10,
  "num_filters": 16,
  "padding": 1,
  "pooling_size": 2,
  "stride": 1,
  "torch_dtype": "float32",
  "transformers_version": "4.30.2"
}