spice-mnist / config.json
rhendz's picture
Upload folder using huggingface_hub (#1)
dc2a7ae
raw
history blame
519 Bytes
{
"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"
}