tzhao3 commited on
Commit
81a507e
1 Parent(s): f8adb35

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -12
config.json CHANGED
@@ -8,21 +8,15 @@
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
- "id2label": {"1": "Blouse",
12
- "2": "Blazer",
13
- "3": "Hoodie",
14
- "4": "Sweater",
15
- "5": "Tank",
16
- "6": "Dress"},
17
  "image_size": 224,
18
  "initializer_range": 0.02,
19
  "intermediate_size": 3072,
20
- "label2id": {"Blouse": "1",
21
- "Blazer": "2",
22
- "Hoodie": "3",
23
- "Sweater": "4",
24
- "Tank": "5",
25
- "Dress": "6"},
26
  "layer_norm_eps": 1e-12,
27
  "model_type": "vit",
28
  "num_attention_heads": 12,
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
+ "id2label": {"1": "upper_body_cloth",
12
+ "2": "lower_body_cloth",
13
+ "3": "full_body_cloth"},
 
 
 
14
  "image_size": 224,
15
  "initializer_range": 0.02,
16
  "intermediate_size": 3072,
17
+ "label2id": {"upper_body_cloth": "1",
18
+ "lower_body_cloth": "2",
19
+ "full_body_cloth": "3"},
 
 
 
20
  "layer_norm_eps": 1e-12,
21
  "model_type": "vit",
22
  "num_attention_heads": 12,