File size: 303 Bytes
8b17fbc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
{
"architectures": [
"SimpleNN"
],
"auto_map": {
"AutoConfig": "configuration_simple_model.SimpleNNConfig",
"AutoModel": "simpleModel.SimpleNN"
},
"input_size": 784,
"model_type": "simple_nn",
"num_classes": 10,
"torch_dtype": "float32",
"transformers_version": "4.44.2"
}
|