| { | |
| "model_type": "custom_resnet", | |
| "auto_map": { | |
| "AutoConfig": "modeling_resnet.ResNetConfig", | |
| "AutoModel": "modeling_resnet.ResNet", | |
| "AutoModelForImageClassification": "modeling_resnet.ResNet" | |
| }, | |
| "architectures": [ | |
| "ResNet" | |
| ], | |
| "num_classes": 10, | |
| "num_channels": 3, | |
| "image_size": 64, | |
| "id2label": { | |
| "0": "0", "1": "1", "2": "2", "3": "3", "4": "4", | |
| "5": "5", "6": "6", "7": "7", "8": "8", "9": "9" | |
| }, | |
| "label2id": { | |
| "0": 0, "1": 1, "2": 2, "3": 3, "4": 4, | |
| "5": 5, "6": 6, "7": 7, "8": 8, "9": 9 | |
| }, | |
| "torch_dtype": "float32" | |
| } |