vladislavbro commited on
Commit
b0184bb
·
verified ·
1 Parent(s): ab68536

Add config from convert_d_fine_original_pytorch_checkpoint_to_hf.py

Browse files
Files changed (1) hide show
  1. config.json +4 -4
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,
@@ -25,6 +23,7 @@
25
  1024,
26
  2048
27
  ],
 
28
  "layer_type": "basic",
29
  "model_type": "hgnet_v2",
30
  "num_channels": 3,
@@ -334,12 +333,13 @@
334
  "positional_encoding_temperature": 10000,
335
  "reg_scale": 4.0,
336
  "top_prob_values": 4,
337
- "torch_dtype": "float32",
338
  "transformers_version": "4.50.0.dev0",
339
  "use_focal_loss": true,
340
  "use_pretrained_backbone": false,
341
  "use_timm_backbone": false,
342
  "weight_loss_bbox": 5.0,
 
 
343
  "weight_loss_giou": 2.0,
344
  "weight_loss_vfl": 1.0,
345
  "with_box_refine": true
 
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,
 
23
  1024,
24
  2048
25
  ],
26
+ "initializer_range": 0.02,
27
  "layer_type": "basic",
28
  "model_type": "hgnet_v2",
29
  "num_channels": 3,
 
333
  "positional_encoding_temperature": 10000,
334
  "reg_scale": 4.0,
335
  "top_prob_values": 4,
 
336
  "transformers_version": "4.50.0.dev0",
337
  "use_focal_loss": true,
338
  "use_pretrained_backbone": false,
339
  "use_timm_backbone": false,
340
  "weight_loss_bbox": 5.0,
341
+ "weight_loss_ddf": 1.5,
342
+ "weight_loss_fgl": 0.15,
343
  "weight_loss_giou": 2.0,
344
  "weight_loss_vfl": 1.0,
345
  "with_box_refine": true