rafaelpadilla commited on
Commit
6f7feee
·
1 Parent(s): e04228d

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +0 -14
config.json CHANGED
@@ -1,9 +1,5 @@
1
  {
2
- "act_decoder": "relu",
3
  "act_encoder": "silu",
4
- "architectures": [
5
- "RTDetrModel"
6
- ],
7
  "aux_loss": true,
8
  "backbone_config": {
9
  "backbone": "resnet50d",
@@ -17,7 +13,6 @@
17
  },
18
  "batch_norm_eps": 1e-05,
19
  "box_noise_scale": 1.0,
20
- "depth_mult": 1.0,
21
  "dim_feedforward": 1024,
22
  "dropout": 0.0,
23
  "encoder_attention_heads": 8,
@@ -28,7 +23,6 @@
28
  640,
29
  640
30
  ],
31
- "expansion": 1.0,
32
  "feat_channels": [
33
  256,
34
  256,
@@ -41,7 +35,6 @@
41
  ],
42
  "focal_loss_alpha": 0.75,
43
  "focal_loss_gamma": 2.0,
44
- "hidden_act_encoder": "gelu",
45
  "hidden_dim": 256,
46
  "id2label": {
47
  "0": "person",
@@ -125,11 +118,6 @@
125
  "8": "boat",
126
  "9": "traffic light"
127
  },
128
- "in_channels": [
129
- 512,
130
- 1024,
131
- 2048
132
- ],
133
  "initializer_range": 0.02,
134
  "label2id": {
135
  "airplane": 4,
@@ -231,8 +219,6 @@
231
  "num_levels": 3,
232
  "num_queries": 300,
233
  "pe_temperature": 10000,
234
- "position_embed_type": "sine",
235
- "torch_dtype": "float32",
236
  "transformers_version": "4.36.0.dev0",
237
  "use_encoder_idx": [
238
  2
 
1
  {
 
2
  "act_encoder": "silu",
 
 
 
3
  "aux_loss": true,
4
  "backbone_config": {
5
  "backbone": "resnet50d",
 
13
  },
14
  "batch_norm_eps": 1e-05,
15
  "box_noise_scale": 1.0,
 
16
  "dim_feedforward": 1024,
17
  "dropout": 0.0,
18
  "encoder_attention_heads": 8,
 
23
  640,
24
  640
25
  ],
 
26
  "feat_channels": [
27
  256,
28
  256,
 
35
  ],
36
  "focal_loss_alpha": 0.75,
37
  "focal_loss_gamma": 2.0,
 
38
  "hidden_dim": 256,
39
  "id2label": {
40
  "0": "person",
 
118
  "8": "boat",
119
  "9": "traffic light"
120
  },
 
 
 
 
 
121
  "initializer_range": 0.02,
122
  "label2id": {
123
  "airplane": 4,
 
219
  "num_levels": 3,
220
  "num_queries": 300,
221
  "pe_temperature": 10000,
 
 
222
  "transformers_version": "4.36.0.dev0",
223
  "use_encoder_idx": [
224
  2