{ | |
"architectures": [ | |
"MnistModel" | |
], | |
"auto_map": { | |
"AutoConfig": "MyConfig.MnistConfig", | |
"AutoModelForImageClassification": "MyModel.MnistModel" | |
}, | |
"conv1": 10, | |
"conv2": 20, | |
"model_type": "MobileNetV1", | |
"torch_dtype": "float32", | |
"transformers_version": "4.38.2" | |
} | |