{ | |
"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" | |
} | |