rafaelpadilla commited on
Commit
68dd376
·
1 Parent(s): 2d00ae0

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -1,9 +1,5 @@
1
  {
2
  "act_encoder": "silu",
3
- "architectures": [
4
- "RTDetrModel"
5
- ],
6
- "aux_loss": true,
7
  "backbone_config": {
8
  "backbone": "resnet50d",
9
  "freeze_batch_norm_2d": true,
@@ -121,6 +117,10 @@
121
  "8": "boat",
122
  "9": "traffic light"
123
  },
 
 
 
 
124
  "initializer_range": 0.02,
125
  "label2id": {
126
  "airplane": 4,
@@ -222,8 +222,8 @@
222
  "num_levels": 3,
223
  "num_queries": 300,
224
  "pe_temperature": 10000,
225
- "torch_dtype": "float32",
226
  "transformers_version": "4.36.0.dev0",
 
227
  "use_encoder_idx": [
228
  2
229
  ],
 
1
  {
2
  "act_encoder": "silu",
 
 
 
 
3
  "backbone_config": {
4
  "backbone": "resnet50d",
5
  "freeze_batch_norm_2d": true,
 
117
  "8": "boat",
118
  "9": "traffic light"
119
  },
120
+ "image_size": [
121
+ 640,
122
+ 640
123
+ ],
124
  "initializer_range": 0.02,
125
  "label2id": {
126
  "airplane": 4,
 
222
  "num_levels": 3,
223
  "num_queries": 300,
224
  "pe_temperature": 10000,
 
225
  "transformers_version": "4.36.0.dev0",
226
+ "use_aux_loss": true,
227
  "use_encoder_idx": [
228
  2
229
  ],