Update config.json

config.json

@@ -3,5 +3,47 @@
   "model_name": "Densenet169_pnuemonia_binaryclassification",
   "architecture": "densenet169",
   "task": "image-classification",
-  "classes": ["Normal", "Pneumonia"]
-}
+  "num_labels": 2,
+  "label2id": {
+    "Normal": 0,
+    "Pneumonia": 1
+  },
+  "id2label": {
+    "0": "Normal",
+    "1": "Pneumonia"
+  },
+  "input_size": [224, 224],
+  "pretrained": true,
+  "weights": "models.DenseNet169_Weights.DEFAULT",
+  "classifier_dropout": 0.4,
+  "batch_size": 128,
+  "epochs": 30,
+  "optimizer": "Lion",
+  "scheduler": "OneCycleLR",
+  "learning_rate": {
+    "features_lr": 7e-5,
+    "classifier_lr": 1e-4
+  },
+  "loss_function": "FocalLoss",
+  "focal_loss_params": {
+    "alpha": 2.5,
+    "gamma": 2
+  },
+  "augmentation": {
+    "resize": [224, 224],
+    "random_flip": 0.3,
+    "random_affine": {
+      "degrees": [-15, 15],
+      "translate": [0.1, 0.1],
+      "scale": [0.85, 1.15]
+    },
+    "random_color_jitter": {
+      "brightness": 0.2,
+      "contrast": 0.2
+    },
+    "random_blur": 0.2,
+    "random_erasing": 0.1
+  },
+  "early_stopping_patience": 7,
+  "device": "cuda"
+}