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 |
}
|