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