Create config.json

config.json

@@ -0,0 +1,44 @@
+{
+  "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"
+}