sbchoi commited on
Commit
dcee887
1 Parent(s): 6ad310c

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,8 +10,9 @@
10
  ],
11
  "attention_dropout": 0.0,
12
  "auxiliary_loss": true,
 
13
  "backbone_config": {
14
- "model_type": "rtdetr-resnet",
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,