sbchoi commited on
Commit
2984e3e
1 Parent(s): 79399fa

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. 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
  "depths": [
15
  3,
@@ -17,7 +18,7 @@
17
  23,
18
  3
19
  ],
20
- "model_type": "rtdetr-resnet",
21
  "out_features": [
22
  "stage2",
23
  "stage3",
@@ -29,6 +30,7 @@
29
  4
30
  ]
31
  },
 
32
  "batch_norm_eps": 1e-05,
33
  "box_noise_scale": 1.0,
34
  "d_model": 256,
@@ -250,6 +252,8 @@
250
  "torch_dtype": "float32",
251
  "transformers_version": "4.42.0.dev0",
252
  "use_focal_loss": true,
 
 
253
  "weight_loss_bbox": 5.0,
254
  "weight_loss_giou": 2.0,
255
  "weight_loss_vfl": 1.0,
 
10
  ],
11
  "attention_dropout": 0.0,
12
  "auxiliary_loss": true,
13
+ "backbone": null,
14
  "backbone_config": {
15
  "depths": [
16
  3,
 
18
  23,
19
  3
20
  ],
21
+ "model_type": "rt_detr_resnet",
22
  "out_features": [
23
  "stage2",
24
  "stage3",
 
30
  4
31
  ]
32
  },
33
+ "backbone_kwargs": null,
34
  "batch_norm_eps": 1e-05,
35
  "box_noise_scale": 1.0,
36
  "d_model": 256,
 
252
  "torch_dtype": "float32",
253
  "transformers_version": "4.42.0.dev0",
254
  "use_focal_loss": true,
255
+ "use_pretrained_backbone": false,
256
+ "use_timm_backbone": false,
257
  "weight_loss_bbox": 5.0,
258
  "weight_loss_giou": 2.0,
259
  "weight_loss_vfl": 1.0,