beingamit99 commited on
Commit
20180c2
1 Parent(s): d8b6f45

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -12
config.json CHANGED
@@ -9,23 +9,23 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "Label_1",
13
- "1": "Label_2",
14
- "2": "Label_3",
15
- "3": "Label_4",
16
- "4": "Label_5",
17
- "5": "Label_6"
18
  },
19
  "image_size": 224,
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "Label_1": 0,
24
- "Label_2": 1,
25
- "Label_3": 2,
26
- "Label_4": 3,
27
- "Label_5": 4,
28
- "Label_6": 5
29
  },
30
  "layer_norm_eps": 1e-12,
31
  "model_type": "vit",
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "Crack",
13
+ "1": "Scratch",
14
+ "2": "Tire Flat",
15
+ "3": "Dent",
16
+ "4": "Glass Shatter",
17
+ "5": "Lamp Broken"
18
  },
19
  "image_size": 224,
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "Crack": 0,
24
+ "Scratch": 1,
25
+ "Tire Flat": 2,
26
+ "Dent": 3,
27
+ "Glass Shatter": 4,
28
+ "Lamp Broken": 5
29
  },
30
  "layer_norm_eps": 1e-12,
31
  "model_type": "vit",