Update config.json
This commit is contained in:
parent
de80c8e5c0
commit
06285c0e36
|
@ -26,9 +26,17 @@
|
||||||
"eos_coefficient": 0.1,
|
"eos_coefficient": 0.1,
|
||||||
"giou_cost": 2,
|
"giou_cost": 2,
|
||||||
"giou_loss_coefficient": 2,
|
"giou_loss_coefficient": 2,
|
||||||
|
"id2label": {
|
||||||
|
"0": "borderless",
|
||||||
|
"1": "bordered"
|
||||||
|
},
|
||||||
"init_std": 0.02,
|
"init_std": 0.02,
|
||||||
"init_xavier_std": 1.0,
|
"init_xavier_std": 1.0,
|
||||||
"is_encoder_decoder": true,
|
"is_encoder_decoder": true,
|
||||||
|
"label2id": {
|
||||||
|
"bordered": 1,
|
||||||
|
"borderless": 0
|
||||||
|
},
|
||||||
"mask_loss_coefficient": 1,
|
"mask_loss_coefficient": 1,
|
||||||
"max_position_embeddings": 1024,
|
"max_position_embeddings": 1024,
|
||||||
"model_type": "detr",
|
"model_type": "detr",
|
||||||
|
|
Loading…
Reference in New Issue