bert-base-uncased-pf-trec / head_config.json
{
"config": {
"activation_function": "tanh",
"bias": true,
"head_type": "classification",
"label2id": {
"DESC": 0,
"ENTY": 1,
"ABBR": 2,
"HUM": 3,
"NUM": 4,
"LOC": 5
},
"layers": 2,
"num_labels": 6,
"use_pooler": false
},
"hidden_size": 768,
"model_class": "BertModelWithHeads",
"model_name": "bert-base-uncased",
"model_type": "bert",
"name": "trec"
}