Add config from convert_relation_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- 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 |
-
"
|
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
|