danelcsb commited on
Commit
0b603b6
·
verified ·
1 Parent(s): 2df155b

Add config from convert_rt_detr_v2_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +3 -4
config.json CHANGED
@@ -3,7 +3,7 @@
3
  "activation_function": "silu",
4
  "anchor_image_size": null,
5
  "architectures": [
6
- "RTDetrForObjectDetection"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": true,
@@ -22,7 +22,7 @@
22
  512
23
  ],
24
  "layer_type": "basic",
25
- "model_type": "rt_detr_resnet",
26
  "out_features": [
27
  "stage2",
28
  "stage3",
@@ -49,7 +49,6 @@
49
  "decoder_layers": 3,
50
  "decoder_n_points": 4,
51
  "decoder_offset_scale": 0.5,
52
- "decoder_version": "v2",
53
  "disable_custom_kernels": true,
54
  "dropout": 0.0,
55
  "encode_proj_layers": [
@@ -251,7 +250,7 @@
251
  "matcher_class_cost": 2.0,
252
  "matcher_gamma": 2.0,
253
  "matcher_giou_cost": 2.0,
254
- "model_type": "rt_detr",
255
  "normalize_before": false,
256
  "num_denoising": 100,
257
  "num_feature_levels": 3,
 
3
  "activation_function": "silu",
4
  "anchor_image_size": null,
5
  "architectures": [
6
+ "RTDetrV2ForObjectDetection"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": true,
 
22
  512
23
  ],
24
  "layer_type": "basic",
25
+ "model_type": "rt_detr_v2_resnet",
26
  "out_features": [
27
  "stage2",
28
  "stage3",
 
49
  "decoder_layers": 3,
50
  "decoder_n_points": 4,
51
  "decoder_offset_scale": 0.5,
 
52
  "disable_custom_kernels": true,
53
  "dropout": 0.0,
54
  "encode_proj_layers": [
 
250
  "matcher_class_cost": 2.0,
251
  "matcher_gamma": 2.0,
252
  "matcher_giou_cost": 2.0,
253
+ "model_type": "rt_detr_v2",
254
  "normalize_before": false,
255
  "num_denoising": 100,
256
  "num_feature_levels": 3,