Grio43 commited on
Commit
1b322b3
·
verified ·
1 Parent(s): 7db3f0b

Add V1_onnx/config.json so HF default download-counting rule applies to ONNX folder pulls

Browse files
Files changed (1) hide show
  1. V1_onnx/config.json +26 -0
V1_onnx/config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "format": "onnx",
3
+ "architecture_type": "vit",
4
+ "num_labels": 19294,
5
+ "num_channels": 3,
6
+ "image_size": 320,
7
+ "patch_size": 16,
8
+ "hidden_size": 1024,
9
+ "num_hidden_layers": 18,
10
+ "num_attention_heads": 16,
11
+ "intermediate_size": 4096,
12
+ "num_groups": 20,
13
+ "tags_per_group": 10000,
14
+ "training_epoch": 33,
15
+ "training_step": 170799,
16
+ "vocab_format_version": 1,
17
+ "vocab_sha256": "b9f95e88fb7e30669077bb761e9a66642ec526c1e10d65336a2a2b628141199d",
18
+ "onnx_opset_version": 21,
19
+ "onnx_inputs": ["pixel_values", "padding_mask"],
20
+ "onnx_outputs": ["probabilities"],
21
+ "dynamic_batch": true,
22
+ "preprocessing_file": "preprocessing.json",
23
+ "thresholds_file": "pr_thresholds.json",
24
+ "vocabulary_file": "vocabulary.json",
25
+ "checkpoint_source": "experiments/run1_vit/checkpoints/last.pt (epoch 33), exported to ONNX"
26
+ }