Create config.json

config.json

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