matejpekar commited on
Commit
f6892c9
·
verified ·
1 Parent(s): f340916

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -2
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "architectures": [
3
  "ViTForImageClassification"
4
  ],
@@ -26,5 +27,15 @@
26
  "pooler_act": "tanh",
27
  "pooler_output_size": 768,
28
  "qkv_bias": true,
29
- "transformers_version": "4.57.6"
30
- }
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "architecture": "vit_base_patch16_224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
27
  "pooler_act": "tanh",
28
  "pooler_output_size": 768,
29
  "qkv_bias": true,
30
+ "transformers_version": "4.57.6",
31
+
32
+ "pretrained_cfg": {
33
+ "architecture": "vit_base_patch16_224",
34
+ "num_classes": 1,
35
+ "input_size": [3, 224, 224],
36
+ "pool_size": null,
37
+ "crop_pct": 0.9,
38
+ "mean": [0.5, 0.5, 0.5],
39
+ "std": [0.5, 0.5, 0.5]
40
+ }
41
+ }