vladislavbro commited on
Commit
afe348e
·
verified ·
1 Parent(s): be482be

Add config from convert_d_fine_original_pytorch_checkpoint_to_hf.py

Browse files
Files changed (1) hide show
  1. config.json +6 -5
config.json CHANGED
@@ -1,10 +1,8 @@
1
  {
 
2
  "activation_dropout": 0.0,
3
  "activation_function": "silu",
4
  "anchor_image_size": null,
5
- "architectures": [
6
- "DFineForObjectDetection"
7
- ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": true,
10
  "backbone": null,
@@ -37,7 +35,11 @@
37
  ],
38
  "decoder_layers": 6,
39
  "decoder_method": "default",
40
- "decoder_n_points": 4,
 
 
 
 
41
  "decoder_offset_scale": 0.5,
42
  "depth_mult": 1.0,
43
  "disable_custom_kernels": true,
@@ -254,7 +256,6 @@
254
  "positional_encoding_temperature": 10000,
255
  "reg_scale": 4.0,
256
  "top_prob_values": 4,
257
- "torch_dtype": "float32",
258
  "transformers_version": "4.50.0.dev0",
259
  "use_focal_loss": true,
260
  "use_pretrained_backbone": false,
 
1
  {
2
+ "_attn_implementation_autoset": true,
3
  "activation_dropout": 0.0,
4
  "activation_function": "silu",
5
  "anchor_image_size": null,
 
 
 
6
  "attention_dropout": 0.0,
7
  "auxiliary_loss": true,
8
  "backbone": null,
 
35
  ],
36
  "decoder_layers": 6,
37
  "decoder_method": "default",
38
+ "decoder_n_points": [
39
+ 3,
40
+ 6,
41
+ 3
42
+ ],
43
  "decoder_offset_scale": 0.5,
44
  "depth_mult": 1.0,
45
  "disable_custom_kernels": true,
 
256
  "positional_encoding_temperature": 10000,
257
  "reg_scale": 4.0,
258
  "top_prob_values": 4,
 
259
  "transformers_version": "4.50.0.dev0",
260
  "use_focal_loss": true,
261
  "use_pretrained_backbone": false,