calpt commited on
Commit
c57c4fa
1 Parent(s): dc8d8d0

Fix label2id mapping.

Browse files
Files changed (1) hide show
  1. head_config.json +3 -3
head_config.json CHANGED
@@ -4,9 +4,9 @@
4
  "bias": true,
5
  "head_type": "classification",
6
  "label2id": {
7
- "LABEL_0": 0,
8
- "LABEL_1": 1,
9
- "LABEL_2": 2
10
  },
11
  "layers": 2,
12
  "num_labels": 3,
 
4
  "bias": true,
5
  "head_type": "classification",
6
  "label2id": {
7
+ "entailment": 0,
8
+ "neutral": 1,
9
+ "contradiction": 2
10
  },
11
  "layers": 2,
12
  "num_labels": 3,