ghaith1997 commited on
Commit
df13232
1 Parent(s): 0d1aa7b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -8
config.json CHANGED
@@ -14,19 +14,19 @@
14
  "hidden_dropout_prob": 0.1,
15
  "hidden_size": 768,
16
  "id2label": {
17
- "0": "LABEL_0",
18
- "1": "LABEL_1",
19
- "2": "LABEL_2",
20
- "3": "LABEL_3"
21
  },
22
  "initializer_range": 0.02,
23
  "input_size": 224,
24
  "intermediate_size": 3072,
25
  "label2id": {
26
- "LABEL_0": 0,
27
- "LABEL_1": 1,
28
- "LABEL_2": 2,
29
- "LABEL_3": 3
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "max_2d_position_embeddings": 1024,
 
14
  "hidden_dropout_prob": 0.1,
15
  "hidden_size": 768,
16
  "id2label": {
17
+ "0": "invoices",
18
+ "1": "email",
19
+ "2": "scientific_publication",
20
+ "3": "resume"
21
  },
22
  "initializer_range": 0.02,
23
  "input_size": 224,
24
  "intermediate_size": 3072,
25
  "label2id": {
26
+ "invoices": 0,
27
+ "email": 1,
28
+ "scientific_publication": 2,
29
+ "resume": 3
30
  },
31
  "layer_norm_eps": 1e-05,
32
  "max_2d_position_embeddings": 1024,