sbchoi commited on
Commit
765cf57
1 Parent(s): 25662e3

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -10,6 +10,7 @@
10
  ],
11
  "attention_dropout": 0.0,
12
  "auxiliary_loss": true,
 
13
  "backbone_config": {
14
  "model_type": "rt_detr_resnet",
15
  "out_features": [
@@ -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": [
 
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,