calpt commited on
Commit
a2dcb37
1 Parent(s): ec64179

Fix label2id mapping.

Browse files

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