rafaelpadilla commited on
Commit
454e576
1 Parent(s): 86d24f3

Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +2 -4
config.json CHANGED
@@ -1,9 +1,6 @@
1
  {
2
  "act_decoder": "relu",
3
  "act_encoder": "silu",
4
- "architectures": [
5
- "RTDetrModel"
6
- ],
7
  "aux_loss": true,
8
  "backbone_config": {
9
  "backbone": "resnet50d",
@@ -213,6 +210,7 @@
213
  "zebra": 22
214
  },
215
  "label_noise_ratio": 0.5,
 
216
  "learnt_init_query": false,
217
  "matcher_alpha": 0.25,
218
  "matcher_bbox_cost": 5.0,
@@ -220,6 +218,7 @@
220
  "matcher_gamma": 2.0,
221
  "matcher_giou_cost": 2.0,
222
  "model_type": "rt_detr",
 
223
  "num_classes": 80,
224
  "num_decoder_layers": 6,
225
  "num_decoder_points": 4,
@@ -230,7 +229,6 @@
230
  "num_queries": 300,
231
  "pe_temperature": 10000,
232
  "position_embed_type": "sine",
233
- "torch_dtype": "float32",
234
  "transformers_version": "4.36.0.dev0",
235
  "use_encoder_idx": [
236
  2
 
1
  {
2
  "act_decoder": "relu",
3
  "act_encoder": "silu",
 
 
 
4
  "aux_loss": true,
5
  "backbone_config": {
6
  "backbone": "resnet50d",
 
210
  "zebra": 22
211
  },
212
  "label_noise_ratio": 0.5,
213
+ "layer_norm_eps": 1e-05,
214
  "learnt_init_query": false,
215
  "matcher_alpha": 0.25,
216
  "matcher_bbox_cost": 5.0,
 
218
  "matcher_gamma": 2.0,
219
  "matcher_giou_cost": 2.0,
220
  "model_type": "rt_detr",
221
+ "normalize_before": false,
222
  "num_classes": 80,
223
  "num_decoder_layers": 6,
224
  "num_decoder_points": 4,
 
229
  "num_queries": 300,
230
  "pe_temperature": 10000,
231
  "position_embed_type": "sine",
 
232
  "transformers_version": "4.36.0.dev0",
233
  "use_encoder_idx": [
234
  2