AntoineBlanot commited on
Commit
99ab01c
1 Parent(s): 259b4b2

Update config.json

Browse files
Files changed (1) hide show
  1. 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": "false",
16
- "1": "true"
 
17
  },
18
  "initializer_factor": 1.0,
19
  "is_encoder_decoder": true,
20
  "is_gated_act": true,
21
  "label2id": {
22
- "false": 0,
23
- "true": 1
 
24
  },
25
  "layer_norm_epsilon": 1e-06,
26
  "model_type": "t5",
27
- "n_class": 2,
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,