sbchoi commited on
Commit
ab23531
1 Parent(s): f64c646

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
  2,
@@ -24,7 +25,7 @@
24
  512
25
  ],
26
  "layer_type": "basic",
27
- "model_type": "rtdetr-resnet",
28
  "out_features": [
29
  "stage2",
30
  "stage3",
@@ -36,6 +37,7 @@
36
  4
37
  ]
38
  },
 
39
  "batch_norm_eps": 1e-05,
40
  "box_noise_scale": 1.0,
41
  "d_model": 256,
@@ -257,6 +259,8 @@
257
  "torch_dtype": "float32",
258
  "transformers_version": "4.42.0.dev0",
259
  "use_focal_loss": true,
 
 
260
  "weight_loss_bbox": 5.0,
261
  "weight_loss_giou": 2.0,
262
  "weight_loss_vfl": 1.0,
 
10
  ],
11
  "attention_dropout": 0.0,
12
  "auxiliary_loss": true,
13
+ "backbone": null,
14
  "backbone_config": {
15
  "depths": [
16
  2,
 
25
  512
26
  ],
27
  "layer_type": "basic",
28
+ "model_type": "rt_detr_resnet",
29
  "out_features": [
30
  "stage2",
31
  "stage3",
 
37
  4
38
  ]
39
  },
40
+ "backbone_kwargs": null,
41
  "batch_norm_eps": 1e-05,
42
  "box_noise_scale": 1.0,
43
  "d_model": 256,
 
259
  "torch_dtype": "float32",
260
  "transformers_version": "4.42.0.dev0",
261
  "use_focal_loss": true,
262
+ "use_pretrained_backbone": false,
263
+ "use_timm_backbone": false,
264
  "weight_loss_bbox": 5.0,
265
  "weight_loss_giou": 2.0,
266
  "weight_loss_vfl": 1.0,