| { | |
| "architectures": ["ResNet18"], | |
| "model_type": "resnet", | |
| "num_classes": 5, | |
| "hidden_sizes": [256, 128, 64], | |
| "image_size": 512, | |
| "num_channels": 3, | |
| "id2label": { | |
| "0": "Normal", | |
| "1": "Mild", | |
| "2": "Moderate", | |
| "3": "Severe", | |
| "4": "Proliferative" | |
| }, | |
| "label2id": { | |
| "Normal": 0, | |
| "Mild": 1, | |
| "Moderate": 2, | |
| "Severe": 3, | |
| "Proliferative": 4 | |
| } | |
| } |