config.json

{
  "architectures": [
    "ResNetForImageClassification"
  ],
  "depth": 50,
  "num_channels": 3,
  "num_labels": 3,
  "id2label": {
    "0": "NORMAL",
    "1": "PNEUMONIA_BACTERIAL",
    "2": "PNEUMONIA_VIRAL"
  },
  "label2id": {
    "NORMAL": 0,
    "PNEUMONIA_BACTERIAL": 1,
    "PNEUMONIA_VIRAL": 2
  },
  "image_size": 224,
  "hidden_sizes": [256, 512, 1024, 2048],
  "model_type": "resnet",
  "torch_dtype": "float32",
  "transformers_version": "4.37.0"
}