Add new model
Browse files- config.json +2 -4
config.json
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
"activation_function": "prelu",
|
| 4 |
"architectures": [
|
| 5 |
-
"
|
| 6 |
],
|
| 7 |
"attention_dropout": 0.0,
|
| 8 |
"auxiliary_loss": false,
|
|
@@ -26,14 +26,12 @@
|
|
| 26 |
"decoder_attention_heads": 8,
|
| 27 |
"decoder_bbox_embed_diff_each_layer": false,
|
| 28 |
"decoder_ffn_dim": 2048,
|
| 29 |
-
"decoder_layerdrop": 0.0,
|
| 30 |
"decoder_layers": 6,
|
| 31 |
"decoder_modulate_hw_attn": true,
|
| 32 |
"do_use_self_attn_decoder": true,
|
| 33 |
"dropout": 0.1,
|
| 34 |
"encoder_attention_heads": 8,
|
| 35 |
"encoder_ffn_dim": 2048,
|
| 36 |
-
"encoder_layerdrop": 0.0,
|
| 37 |
"encoder_layers": 6,
|
| 38 |
"focal_alpha": 0.25,
|
| 39 |
"giou_cost": 2,
|
|
@@ -234,7 +232,7 @@
|
|
| 234 |
"temperature_height": 20,
|
| 235 |
"temperature_width": 20,
|
| 236 |
"torch_dtype": "float32",
|
| 237 |
-
"transformers_version": "4.
|
| 238 |
"use_pretrained_backbone": true,
|
| 239 |
"use_timm_backbone": true
|
| 240 |
}
|
|
|
|
| 2 |
"activation_dropout": 0.0,
|
| 3 |
"activation_function": "prelu",
|
| 4 |
"architectures": [
|
| 5 |
+
"DabDetrForObjectDetection"
|
| 6 |
],
|
| 7 |
"attention_dropout": 0.0,
|
| 8 |
"auxiliary_loss": false,
|
|
|
|
| 26 |
"decoder_attention_heads": 8,
|
| 27 |
"decoder_bbox_embed_diff_each_layer": false,
|
| 28 |
"decoder_ffn_dim": 2048,
|
|
|
|
| 29 |
"decoder_layers": 6,
|
| 30 |
"decoder_modulate_hw_attn": true,
|
| 31 |
"do_use_self_attn_decoder": true,
|
| 32 |
"dropout": 0.1,
|
| 33 |
"encoder_attention_heads": 8,
|
| 34 |
"encoder_ffn_dim": 2048,
|
|
|
|
| 35 |
"encoder_layers": 6,
|
| 36 |
"focal_alpha": 0.25,
|
| 37 |
"giou_cost": 2,
|
|
|
|
| 232 |
"temperature_height": 20,
|
| 233 |
"temperature_width": 20,
|
| 234 |
"torch_dtype": "float32",
|
| 235 |
+
"transformers_version": "4.46.0.dev0",
|
| 236 |
"use_pretrained_backbone": true,
|
| 237 |
"use_timm_backbone": true
|
| 238 |
}
|