roberta-base-pf-race / head_config.json
calpt's picture
Fix label2id mapping.
b581ae5
raw
history blame contribute delete
380 Bytes
{
"config": {
"activation_function": "tanh",
"head_type": "multiple_choice",
"label2id": {
"A": 0,
"B": 1,
"C": 2,
"D": 3
},
"layers": 2,
"num_choices": 4,
"use_pooler": false
},
"hidden_size": 768,
"model_class": "RobertaModelWithHeads",
"model_name": "roberta-base",
"model_type": "roberta",
"name": "race_all"
}