config.json

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