config.json
587 B · 33 lines · json Raw
1 {
2 "architecture": "vit_small_patch16_224",
3 "num_classes": 1000,
4 "num_features": 384,
5 "global_pool": "token",
6 "pretrained_cfg": {
7 "tag": "augreg_in21k_ft_in1k",
8 "custom_load": true,
9 "input_size": [
10 3,
11 224,
12 224
13 ],
14 "fixed_input_size": true,
15 "interpolation": "bicubic",
16 "crop_pct": 0.9,
17 "crop_mode": "center",
18 "mean": [
19 0.5,
20 0.5,
21 0.5
22 ],
23 "std": [
24 0.5,
25 0.5,
26 0.5
27 ],
28 "num_classes": 1000,
29 "pool_size": null,
30 "first_conv": "patch_embed.proj",
31 "classifier": "head"
32 }
33 }