calpt commited on
Commit
b5fe62b
1 Parent(s): b89aeab

Fix label2id mapping.

Browse files

Files changed (1) hide show
  1. head_config.json +3 -3
head_config.json CHANGED
@@ -3,9 +3,9 @@
3
  "activation_function": "tanh",
4
  "head_type": "multiple_choice",
5
  "label2id": {
6
- "LABEL_0": 0,
7
- "LABEL_1": 1,
8
- "LABEL_2": 2
9
  },
10
  "layers": 2,
11
  "num_choices": 3,
3
  "activation_function": "tanh",
4
  "head_type": "multiple_choice",
5
  "label2id": {
6
+ "1": 0,
7
+ "2": 1,
8
+ "3": 2
9
  },
10
  "layers": 2,
11
  "num_choices": 3,