From 06285c0e36603bca9ac767818ce6025eb53039d5 Mon Sep 17 00:00:00 2001
From: Taha Douaji <taha.d@alpha-apps.ae>
Date: Fri, 11 Mar 2022 22:41:44 +0000
Subject: [PATCH] Update config.json

---
 config.json | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/config.json b/config.json
index adcd723..624592e 100644
--- a/config.json
+++ b/config.json
@@ -26,9 +26,17 @@
   "eos_coefficient": 0.1,
   "giou_cost": 2,
   "giou_loss_coefficient": 2,
+  "id2label": {
+    "0": "borderless",
+    "1": "bordered"
+  },
   "init_std": 0.02,
   "init_xavier_std": 1.0,
   "is_encoder_decoder": true,
+  "label2id": {
+    "bordered": 1,
+    "borderless": 0
+  },
   "mask_loss_coefficient": 1,
   "max_position_embeddings": 1024,
   "model_type": "detr",