vladislavbro commited on
Commit
12e0bff
·
verified ·
1 Parent(s): f797a2e

Add config from convert_d_fine_original_pytorch_checkpoint_to_hf.py

Browse files
Files changed (1) hide show
  1. config.json +3 -6
config.json CHANGED
@@ -1,20 +1,18 @@
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,
11
  "backbone_config": {
12
  "embedding_size": 32,
13
  "hidden_sizes": [
14
- 64,
15
  128,
16
  256,
17
- 512
 
18
  ],
19
  "layer_type": "basic",
20
  "model_type": "d_fine_resnet",
@@ -308,7 +306,6 @@
308
  "reg_max": 32,
309
  "reg_scale": 4.0,
310
  "top_prob_values": 4,
311
- "torch_dtype": "float32",
312
  "transformers_version": "4.49.0.dev0",
313
  "use_focal_loss": true,
314
  "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,
9
  "backbone_config": {
10
  "embedding_size": 32,
11
  "hidden_sizes": [
 
12
  128,
13
  256,
14
+ 512,
15
+ 1024
16
  ],
17
  "layer_type": "basic",
18
  "model_type": "d_fine_resnet",
 
306
  "reg_max": 32,
307
  "reg_scale": 4.0,
308
  "top_prob_values": 4,
 
309
  "transformers_version": "4.49.0.dev0",
310
  "use_focal_loss": true,
311
  "use_pretrained_backbone": false,