| { | |
| "architectures": ["CervicalCancerCNN"], | |
| "model_type": "cervical-cancer-cnn", | |
| "auto_map": { | |
| "AutoModel": "modeling_cervical.CervicalCancerCNN" | |
| }, | |
| "num_labels": 4, | |
| "num_classes": 4, | |
| "id2label": { | |
| "0": "Normal", | |
| "1": "LSIL", | |
| "2": "HSIL", | |
| "3": "Cancer" | |
| }, | |
| "label2id": { | |
| "Normal": 0, | |
| "LSIL": 1, | |
| "HSIL": 2, | |
| "Cancer": 3 | |
| }, | |
| "conv_layers": [32, 64, 128, 256], | |
| "fc_layers": [256, 128], | |
| "dropout": 0.5, | |
| "input_channels": 3, | |
| "input_size": { | |
| "height": 224, | |
| "width": 298 | |
| }, | |
| "total_parameters": 488580, | |
| "problem_type": "single_label_classification", | |
| "torch_dtype": "float32" | |
| } | |