vladislavbro commited on
Commit
f797a2e
·
verified ·
1 Parent(s): 0e166cc

Add model from convert_d_fine_original_pytorch_checkpoint_to_hf.py

Browse files
Files changed (2) hide show
  1. config.json +4 -1
  2. model.safetensors +2 -2
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,
@@ -306,6 +308,7 @@
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,
 
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,
 
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,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c82cb82a0a55c4a20e862cf71e535b30ac7bbab445443a5ba9d60d3cb2cb4c50
3
- size 251619448
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f99dfc5ef643715c091084616053ae4c85de568a64feb8c51afa7a1690f08e8
3
+ size 41497668