Update config.json

config.json

@@ -16,5 +16,29 @@
     "labels": [
         {"name": "Architrave", "attributes": []},
         {"name": "Doojam", "attributes": []}
-    ]
+    ],
+    "id2label": {
+        "0": "Architrave",
+        "1": "Doojam"
+    },
+    "hyperparameters": {
+        "epochs": 100,
+        "imgsz": 640,
+        "batch_size": 16,
+        "learning_rate": 0.01,
+        "momentum": 0.9,
+        "weight_decay": 0.0005,
+        "optimizer": "SGD",
+        "lr_scheduler": "StepLR",
+        "step_size": 10,
+        "gamma": 0.1
+    },
+    "architectures": ["YOLOv8"],
+    "model_name_or_path": "ultralytics/yolov8n-seg",
+    "use_cache": false,
+    "do_sample": false,
+    "early_stopping": true,
+    "fp16": false,
+    "num_beams": 1,
+    "use_legacy_prediction_head": false
 }