AntoineBlanot
commited on
Commit
•
99ab01c
1
Parent(s):
259b4b2
Update config.json
Browse files- config.json +7 -5
config.json
CHANGED
@@ -12,19 +12,21 @@
|
|
12 |
"eos_token_id": 1,
|
13 |
"feed_forward_proj": "gated-gelu",
|
14 |
"id2label": {
|
15 |
-
"0": "
|
16 |
-
"1": "
|
|
|
17 |
},
|
18 |
"initializer_factor": 1.0,
|
19 |
"is_encoder_decoder": true,
|
20 |
"is_gated_act": true,
|
21 |
"label2id": {
|
22 |
-
"
|
23 |
-
"
|
|
|
24 |
},
|
25 |
"layer_norm_epsilon": 1e-06,
|
26 |
"model_type": "t5",
|
27 |
-
"n_class":
|
28 |
"n_positions": 512,
|
29 |
"num_decoder_layers": 24,
|
30 |
"num_heads": 16,
|
|
|
12 |
"eos_token_id": 1,
|
13 |
"feed_forward_proj": "gated-gelu",
|
14 |
"id2label": {
|
15 |
+
"0": "entailment",
|
16 |
+
"1": "neutral",
|
17 |
+
"2": "contradiction"
|
18 |
},
|
19 |
"initializer_factor": 1.0,
|
20 |
"is_encoder_decoder": true,
|
21 |
"is_gated_act": true,
|
22 |
"label2id": {
|
23 |
+
"entailment": 0,
|
24 |
+
"neutral": 1,
|
25 |
+
"contradiction": 2
|
26 |
},
|
27 |
"layer_norm_epsilon": 1e-06,
|
28 |
"model_type": "t5",
|
29 |
+
"n_class": 3,
|
30 |
"n_positions": 512,
|
31 |
"num_decoder_layers": 24,
|
32 |
"num_heads": 16,
|