vladislavbro commited on
Commit
5446de0
·
verified ·
1 Parent(s): 55891fe

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,
@@ -309,6 +311,7 @@
309
  "reg_max": 32,
310
  "reg_scale": 4.0,
311
  "top_prob_values": 4,
 
312
  "transformers_version": "4.49.0.dev0",
313
  "use_focal_loss": true,
314
  "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,
 
311
  "reg_max": 32,
312
  "reg_scale": 4.0,
313
  "top_prob_values": 4,
314
+ "torch_dtype": "float32",
315
  "transformers_version": "4.49.0.dev0",
316
  "use_focal_loss": true,
317
  "use_pretrained_backbone": false,