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",