xiuqhou commited on
Commit
abf4ca6
·
verified ·
1 Parent(s): eb8051e

Add config from convert_relation_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +5 -8
config.json CHANGED
@@ -1,14 +1,12 @@
1
  {
 
2
  "activation_dropout": 0.0,
3
  "activation_function": "relu",
4
- "architectures": [
5
- "RelationDetrForObjectDetection"
6
- ],
7
  "attention_dropout": 0.0,
8
  "backbone": "resnet50",
9
  "backbone_config": null,
 
10
  "backbone_kwargs": {
11
- "in_chans": 3,
12
  "out_indices": [
13
  2,
14
  3,
@@ -27,7 +25,6 @@
27
  "decoder_ffn_dim": 2048,
28
  "decoder_layers": 6,
29
  "decoder_n_points": 4,
30
- "dilation": false,
31
  "disable_custom_kernels": false,
32
  "dropout": 0.0,
33
  "encoder_attention_heads": 8,
@@ -133,6 +130,7 @@
133
  "89": "hair drier",
134
  "90": "toothbrush"
135
  },
 
136
  "init_std": 0.02,
137
  "init_xavier_std": 1.0,
138
  "is_encoder_decoder": true,
@@ -220,8 +218,8 @@
220
  "zebra": 24
221
  },
222
  "label_noise_ratio": 0.5,
 
223
  "model_type": "relation_detr",
224
- "num_channels": 3,
225
  "num_denoising": 100,
226
  "num_feature_levels": 4,
227
  "num_queries": 900,
@@ -231,8 +229,7 @@
231
  "sin_cos_offset": -0.5,
232
  "sin_cos_scale": 6.283185307179586,
233
  "sin_cos_temperature": 10000,
234
- "torch_dtype": "float32",
235
- "transformers_version": "4.47.0.dev0",
236
  "two_stage_binary_cls": false,
237
  "use_pretrained_backbone": false,
238
  "use_timm_backbone": true
 
1
  {
2
+ "_attn_implementation_autoset": true,
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
 
 
 
5
  "attention_dropout": 0.0,
6
  "backbone": "resnet50",
7
  "backbone_config": null,
8
+ "backbone_features_format": "channels_first",
9
  "backbone_kwargs": {
 
10
  "out_indices": [
11
  2,
12
  3,
 
25
  "decoder_ffn_dim": 2048,
26
  "decoder_layers": 6,
27
  "decoder_n_points": 4,
 
28
  "disable_custom_kernels": false,
29
  "dropout": 0.0,
30
  "encoder_attention_heads": 8,
 
130
  "89": "hair drier",
131
  "90": "toothbrush"
132
  },
133
+ "init_bias_prior_prob": null,
134
  "init_std": 0.02,
135
  "init_xavier_std": 1.0,
136
  "is_encoder_decoder": true,
 
218
  "zebra": 24
219
  },
220
  "label_noise_ratio": 0.5,
221
+ "layer_norm_eps": 1e-05,
222
  "model_type": "relation_detr",
 
223
  "num_denoising": 100,
224
  "num_feature_levels": 4,
225
  "num_queries": 900,
 
229
  "sin_cos_offset": -0.5,
230
  "sin_cos_scale": 6.283185307179586,
231
  "sin_cos_temperature": 10000,
232
+ "transformers_version": "4.48.0.dev0",
 
233
  "two_stage_binary_cls": false,
234
  "use_pretrained_backbone": false,
235
  "use_timm_backbone": true