Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- config.json +5 -1
config.json
CHANGED
|
@@ -10,6 +10,7 @@
|
|
| 10 |
],
|
| 11 |
"attention_dropout": 0.0,
|
| 12 |
"auxiliary_loss": true,
|
|
|
|
| 13 |
"backbone_config": {
|
| 14 |
"hidden_sizes": [
|
| 15 |
64,
|
|
@@ -18,7 +19,7 @@
|
|
| 18 |
512
|
| 19 |
],
|
| 20 |
"layer_type": "basic",
|
| 21 |
-
"model_type": "
|
| 22 |
"out_features": [
|
| 23 |
"stage2",
|
| 24 |
"stage3",
|
|
@@ -30,6 +31,7 @@
|
|
| 30 |
4
|
| 31 |
]
|
| 32 |
},
|
|
|
|
| 33 |
"batch_norm_eps": 1e-05,
|
| 34 |
"box_noise_scale": 1.0,
|
| 35 |
"d_model": 256,
|
|
@@ -251,6 +253,8 @@
|
|
| 251 |
"torch_dtype": "float32",
|
| 252 |
"transformers_version": "4.42.0.dev0",
|
| 253 |
"use_focal_loss": true,
|
|
|
|
|
|
|
| 254 |
"weight_loss_bbox": 5.0,
|
| 255 |
"weight_loss_giou": 2.0,
|
| 256 |
"weight_loss_vfl": 1.0,
|
|
|
|
| 10 |
],
|
| 11 |
"attention_dropout": 0.0,
|
| 12 |
"auxiliary_loss": true,
|
| 13 |
+
"backbone": null,
|
| 14 |
"backbone_config": {
|
| 15 |
"hidden_sizes": [
|
| 16 |
64,
|
|
|
|
| 19 |
512
|
| 20 |
],
|
| 21 |
"layer_type": "basic",
|
| 22 |
+
"model_type": "rt_detr_resnet",
|
| 23 |
"out_features": [
|
| 24 |
"stage2",
|
| 25 |
"stage3",
|
|
|
|
| 31 |
4
|
| 32 |
]
|
| 33 |
},
|
| 34 |
+
"backbone_kwargs": null,
|
| 35 |
"batch_norm_eps": 1e-05,
|
| 36 |
"box_noise_scale": 1.0,
|
| 37 |
"d_model": 256,
|
|
|
|
| 253 |
"torch_dtype": "float32",
|
| 254 |
"transformers_version": "4.42.0.dev0",
|
| 255 |
"use_focal_loss": true,
|
| 256 |
+
"use_pretrained_backbone": false,
|
| 257 |
+
"use_timm_backbone": false,
|
| 258 |
"weight_loss_bbox": 5.0,
|
| 259 |
"weight_loss_giou": 2.0,
|
| 260 |
"weight_loss_vfl": 1.0,
|