Andrazp commited on
Commit
cb28a19
1 Parent(s): 88e8da9

changed config to enable loading with automodel

Browse files
Files changed (1) hide show
  1. config.json +13 -0
config.json CHANGED
@@ -8,11 +8,24 @@
8
  "hidden_dropout_prob": 0.1,
9
  "hidden_size": 768,
10
  "initializer_range": 0.02,
 
 
 
 
 
 
11
  "intermediate_size": 3072,
 
 
 
 
 
 
12
  "is_decoder": false,
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
15
  "model_type": "bert",
 
16
  "num_attention_heads": 12,
17
  "num_hidden_layers": 12,
18
  "num_labels": 4,
 
8
  "hidden_dropout_prob": 0.1,
9
  "hidden_size": 768,
10
  "initializer_range": 0.02,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3"
16
+ },
17
  "intermediate_size": 3072,
18
+ "label2id": {
19
+ "LABEL_0": 0,
20
+ "LABEL_1": 1,
21
+ "LABEL_2": 2,
22
+ "LABEL_3": 3
23
+ },
24
  "is_decoder": false,
25
  "layer_norm_eps": 1e-12,
26
  "max_position_embeddings": 512,
27
  "model_type": "bert",
28
+ "model_type": "bert",
29
  "num_attention_heads": 12,
30
  "num_hidden_layers": 12,
31
  "num_labels": 4,