vladislavbro commited on
Commit
aaf0d55
·
verified ·
1 Parent(s): e9187c2

Add model from convert_d_fine_original_pytorch_checkpoint_to_hf.py

Browse files
Files changed (1) hide show
  1. config.json +4 -1
config.json CHANGED
@@ -1,8 +1,10 @@
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,
@@ -302,6 +304,7 @@
302
  "positional_encoding_temperature": 10000,
303
  "reg_max": 32,
304
  "reg_scale": 4.0,
 
305
  "transformers_version": "4.49.0.dev0",
306
  "use_focal_loss": true,
307
  "use_pretrained_backbone": false,
 
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,
 
304
  "positional_encoding_temperature": 10000,
305
  "reg_max": 32,
306
  "reg_scale": 4.0,
307
+ "torch_dtype": "float32",
308
  "transformers_version": "4.49.0.dev0",
309
  "use_focal_loss": true,
310
  "use_pretrained_backbone": false,