TahaDouaji
commited on
Commit
•
06285c0
1
Parent(s):
de80c8e
Update config.json
Browse files- config.json +8 -0
config.json
CHANGED
@@ -26,9 +26,17 @@
|
|
26 |
"eos_coefficient": 0.1,
|
27 |
"giou_cost": 2,
|
28 |
"giou_loss_coefficient": 2,
|
|
|
|
|
|
|
|
|
29 |
"init_std": 0.02,
|
30 |
"init_xavier_std": 1.0,
|
31 |
"is_encoder_decoder": true,
|
|
|
|
|
|
|
|
|
32 |
"mask_loss_coefficient": 1,
|
33 |
"max_position_embeddings": 1024,
|
34 |
"model_type": "detr",
|
|
|
26 |
"eos_coefficient": 0.1,
|
27 |
"giou_cost": 2,
|
28 |
"giou_loss_coefficient": 2,
|
29 |
+
"id2label": {
|
30 |
+
"0": "borderless",
|
31 |
+
"1": "bordered"
|
32 |
+
},
|
33 |
"init_std": 0.02,
|
34 |
"init_xavier_std": 1.0,
|
35 |
"is_encoder_decoder": true,
|
36 |
+
"label2id": {
|
37 |
+
"bordered": 1,
|
38 |
+
"borderless": 0
|
39 |
+
},
|
40 |
"mask_loss_coefficient": 1,
|
41 |
"max_position_embeddings": 1024,
|
42 |
"model_type": "detr",
|