Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- config.json +5 -1
config.json
CHANGED
@@ -10,8 +10,9 @@
|
|
10 |
],
|
11 |
"attention_dropout": 0.0,
|
12 |
"auxiliary_loss": true,
|
|
|
13 |
"backbone_config": {
|
14 |
-
"model_type": "
|
15 |
"out_features": [
|
16 |
"stage2",
|
17 |
"stage3",
|
@@ -23,6 +24,7 @@
|
|
23 |
4
|
24 |
]
|
25 |
},
|
|
|
26 |
"batch_norm_eps": 1e-05,
|
27 |
"box_noise_scale": 1.0,
|
28 |
"d_model": 256,
|
@@ -244,6 +246,8 @@
|
|
244 |
"torch_dtype": "float32",
|
245 |
"transformers_version": "4.42.0.dev0",
|
246 |
"use_focal_loss": true,
|
|
|
|
|
247 |
"weight_loss_bbox": 5.0,
|
248 |
"weight_loss_giou": 2.0,
|
249 |
"weight_loss_vfl": 1.0,
|
|
|
10 |
],
|
11 |
"attention_dropout": 0.0,
|
12 |
"auxiliary_loss": true,
|
13 |
+
"backbone": null,
|
14 |
"backbone_config": {
|
15 |
+
"model_type": "rt_detr_resnet",
|
16 |
"out_features": [
|
17 |
"stage2",
|
18 |
"stage3",
|
|
|
24 |
4
|
25 |
]
|
26 |
},
|
27 |
+
"backbone_kwargs": null,
|
28 |
"batch_norm_eps": 1e-05,
|
29 |
"box_noise_scale": 1.0,
|
30 |
"d_model": 256,
|
|
|
246 |
"torch_dtype": "float32",
|
247 |
"transformers_version": "4.42.0.dev0",
|
248 |
"use_focal_loss": true,
|
249 |
+
"use_pretrained_backbone": false,
|
250 |
+
"use_timm_backbone": false,
|
251 |
"weight_loss_bbox": 5.0,
|
252 |
"weight_loss_giou": 2.0,
|
253 |
"weight_loss_vfl": 1.0,
|