matejpekar commited on
Commit
3924fb9
·
verified ·
1 Parent(s): e886b29

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -4
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "architecture": "vit_base_patch16_224.augreg_in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -30,11 +30,10 @@
30
  "transformers_version": "4.57.6",
31
 
32
  "pretrained_cfg": {
33
- "architecture": "vit_base_patch16_224.augreg_in21k",
 
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
  }
 
1
  {
2
+ "architecture": "vit_base_patch16_224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
30
  "transformers_version": "4.57.6",
31
 
32
  "pretrained_cfg": {
33
+ "architecture": "vit_base_patch16_224",
34
+ "file": "timm_model.bin",
35
  "num_classes": 1,
36
  "input_size": [3, 224, 224],
 
 
37
  "mean": [0.5, 0.5, 0.5],
38
  "std": [0.5, 0.5, 0.5]
39
  }