rafaelpadilla commited on
Commit
f30cbbc
·
1 Parent(s): 84cde37

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +3 -8
config.json CHANGED
@@ -1,8 +1,5 @@
1
  {
2
  "act_encoder": "silu",
3
- "architectures": [
4
- "RTDetrModel"
5
- ],
6
  "backbone_config": {
7
  "backbone": "resnet50d",
8
  "freeze_batch_norm_2d": true,
@@ -17,6 +14,9 @@
17
  "box_noise_scale": 1.0,
18
  "dim_feedforward": 1024,
19
  "dropout": 0.0,
 
 
 
20
  "encoder_attention_heads": 8,
21
  "eos_coefficient": 0.1,
22
  "eval_idx": -1,
@@ -217,7 +217,6 @@
217
  "matcher_giou_cost": 2.0,
218
  "model_type": "rt_detr",
219
  "normalize_before": false,
220
- "num_classes": 80,
221
  "num_decoder_layers": 6,
222
  "num_decoder_points": 4,
223
  "num_denoising": 100,
@@ -225,12 +224,8 @@
225
  "num_levels": 3,
226
  "num_queries": 300,
227
  "pe_temperature": 10000,
228
- "torch_dtype": "float32",
229
  "transformers_version": "4.36.0.dev0",
230
  "use_aux_loss": true,
231
- "use_encoder_idx": [
232
- 2
233
- ],
234
  "use_focal_loss": true,
235
  "weight_loss_bbox": 5.0,
236
  "weight_loss_giou": 2.0,
 
1
  {
2
  "act_encoder": "silu",
 
 
 
3
  "backbone_config": {
4
  "backbone": "resnet50d",
5
  "freeze_batch_norm_2d": true,
 
14
  "box_noise_scale": 1.0,
15
  "dim_feedforward": 1024,
16
  "dropout": 0.0,
17
+ "encode_proj_layers": [
18
+ 2
19
+ ],
20
  "encoder_attention_heads": 8,
21
  "eos_coefficient": 0.1,
22
  "eval_idx": -1,
 
217
  "matcher_giou_cost": 2.0,
218
  "model_type": "rt_detr",
219
  "normalize_before": false,
 
220
  "num_decoder_layers": 6,
221
  "num_decoder_points": 4,
222
  "num_denoising": 100,
 
224
  "num_levels": 3,
225
  "num_queries": 300,
226
  "pe_temperature": 10000,
 
227
  "transformers_version": "4.36.0.dev0",
228
  "use_aux_loss": true,
 
 
 
229
  "use_focal_loss": true,
230
  "weight_loss_bbox": 5.0,
231
  "weight_loss_giou": 2.0,