gacfox commited on
Commit
a63464c
·
verified ·
1 Parent(s): 6c3b6ff

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +40 -0
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "swin",
3
+ "num_classes": 5,
4
+ "num_features": 1024,
5
+ "global_pool": "avg",
6
+ "mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "input_size": [
17
+ 3,
18
+ 224,
19
+ 224
20
+ ],
21
+ "interpolation": "bicubic",
22
+ "crop_pct": 0.9,
23
+ "crop_mode": "center",
24
+ "first_conv": "patch_embed.proj",
25
+ "classifier": "head",
26
+ "id2label": {
27
+ "0": "content",
28
+ "1": "end",
29
+ "2": "start",
30
+ "3": "subt",
31
+ "4": "subtl"
32
+ },
33
+ "label2id": {
34
+ "content": 0,
35
+ "end": 1,
36
+ "start": 2,
37
+ "subt": 3,
38
+ "subtl": 4
39
+ }
40
+ }