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