{
  "model_type": "googlenet",
  "model_name": "Googlenet_pneumonia_binaryclassification",
  "architecture": "googlenet",
  "task": "image-classification",
  "num_labels": 2,
  "label2id": {
    "Normal": 0,
    "Pneumonia": 1
  },
  "id2label": {
    "0": "Normal",
    "1": "Pneumonia"
  },
  "input_size": [224, 224],
  "pretrained": true,
  "weights": "models.GoogLeNet_Weights.DEFAULT",
  "classifier_dropout": 0.5,
  "batch_size": 256,
  "epochs": 20,
  "optimizer": "Adam",
  "scheduler": "ReduceLROnPlateau",
  "learning_rate": {
    "features_lr": 1e-4,
    "classifier_lr": 1e-4
  },
  "loss_function": "CrossEntropyLoss",
  "label_smoothing": 0.1,
  "augmentation": {
    "resize": [224, 224],
    "random_flip": 0.3,
    "random_affine": {
      "degrees": [-5, 5],
      "translate": [0.05, 0.05],
      "scale": [0.95, 1.05]
    },
    "random_color_jitter": {
      "brightness": 0.2,
      "contrast": 0.2
    }
  },
  "early_stopping_patience": 3,
  "device": "cuda"
}