config.json

{
  "model_type": "resnet",
  "model_name": "Resnet18_pnuemonia_binaryclassification",
  "architecture": "resnet18",
  "task": "image-classification",
  "num_labels": 2,
  "label2id": {
    "Normal": 0,
    "Pneumonia": 1
  },
  "id2label": {
    "0": "Normal",
    "1": "Pneumonia"
  },
  "input_size": [224, 224],
  "pretrained": true,
  "weights": "models.ResNet18_Weights.DEFAULT",
  "classifier_dropout": 0.5,
  "batch_size": 512,
  "epochs": 25,
  "optimizer": "AdamW",
  "optimizer_params": {
    "lr": 1e-4,
    "betas": [0.9, 0.999],
    "eps": 1e-8,
    "weight_decay": 1e-4
  },
  "scheduler": {
    "type": "OneCycleLR",
    "max_lr": [1e-4, 1e-4, 1e-4, 1e-4, 1e-4, 1e-3],
    "epochs": 25,
    "pct_start": 0.3,
    "anneal_strategy": "cos"
  },
  "loss_function": "FocalLoss",
  "focal_loss_params": {
    "alpha": 1,
    "gamma": 2
  },
  "augmentation": {
    "resize": [224, 224],
    "random_flip": 0.5,
    "random_affine": {
      "degrees": [-10, 10],
      "translate": [0.1, 0.1],
      "scale": [0.9, 1.1]
    },
    "color_jitter": {
      "brightness": 0.3,
      "contrast": 0.3
    },
    "random_blur": 0.2,
    "random_erasing": 0.2
  },
  "early_stopping_patience": 5,
  "device": "cuda"
}