Update config.json
Browse files- config.json +6 -10
config.json
CHANGED
@@ -7,18 +7,17 @@
|
|
7 |
"attention_dropout": 0.0,
|
8 |
"auxiliary_loss": false,
|
9 |
"backbone": "resnet50",
|
10 |
-
"backbone_config": null,
|
11 |
"bbox_cost": 5,
|
12 |
"bbox_loss_coefficient": 5,
|
13 |
"class_cost": 1,
|
14 |
"d_model": 256,
|
15 |
"decoder_attention_heads": 8,
|
16 |
"decoder_ffn_dim": 1024,
|
|
|
17 |
"decoder_layers": 6,
|
18 |
"decoder_n_points": 4,
|
19 |
"dice_loss_coefficient": 1,
|
20 |
"dilation": false,
|
21 |
-
"disable_custom_kernels": false,
|
22 |
"dropout": 0.1,
|
23 |
"encoder_attention_heads": 8,
|
24 |
"encoder_ffn_dim": 1024,
|
@@ -26,7 +25,6 @@
|
|
26 |
"encoder_layers": 6,
|
27 |
"encoder_n_points": 4,
|
28 |
"eos_coefficient": 0.1,
|
29 |
-
"focal_alpha": 0.25,
|
30 |
"giou_cost": 2,
|
31 |
"giou_loss_coefficient": 2,
|
32 |
"id2label": {
|
@@ -63,15 +61,13 @@
|
|
63 |
"mask_loss_coefficient": 1,
|
64 |
"max_position_embeddings": 1024,
|
65 |
"model_type": "deformable_detr",
|
66 |
-
"num_channels": 3,
|
67 |
"num_feature_levels": 4,
|
68 |
-
"num_queries":
|
69 |
"position_embedding_type": "sine",
|
|
|
70 |
"torch_dtype": "float32",
|
71 |
-
"transformers_version": "4.
|
72 |
-
"two_stage":
|
73 |
-
"two_stage_num_proposals":
|
74 |
-
"use_pretrained_backbone": true,
|
75 |
-
"use_timm_backbone": "SenseTime/deformable-detr-with-box-refine-two-stage",
|
76 |
"with_box_refine": true
|
77 |
}
|
|
|
7 |
"attention_dropout": 0.0,
|
8 |
"auxiliary_loss": false,
|
9 |
"backbone": "resnet50",
|
|
|
10 |
"bbox_cost": 5,
|
11 |
"bbox_loss_coefficient": 5,
|
12 |
"class_cost": 1,
|
13 |
"d_model": 256,
|
14 |
"decoder_attention_heads": 8,
|
15 |
"decoder_ffn_dim": 1024,
|
16 |
+
"decoder_layerdrop": 0.0,
|
17 |
"decoder_layers": 6,
|
18 |
"decoder_n_points": 4,
|
19 |
"dice_loss_coefficient": 1,
|
20 |
"dilation": false,
|
|
|
21 |
"dropout": 0.1,
|
22 |
"encoder_attention_heads": 8,
|
23 |
"encoder_ffn_dim": 1024,
|
|
|
25 |
"encoder_layers": 6,
|
26 |
"encoder_n_points": 4,
|
27 |
"eos_coefficient": 0.1,
|
|
|
28 |
"giou_cost": 2,
|
29 |
"giou_loss_coefficient": 2,
|
30 |
"id2label": {
|
|
|
61 |
"mask_loss_coefficient": 1,
|
62 |
"max_position_embeddings": 1024,
|
63 |
"model_type": "deformable_detr",
|
|
|
64 |
"num_feature_levels": 4,
|
65 |
+
"num_queries": 300,
|
66 |
"position_embedding_type": "sine",
|
67 |
+
"return_intermediate": true,
|
68 |
"torch_dtype": "float32",
|
69 |
+
"transformers_version": "4.16.0.dev0",
|
70 |
+
"two_stage": false,
|
71 |
+
"two_stage_num_proposals": 300,
|
|
|
|
|
72 |
"with_box_refine": true
|
73 |
}
|