File size: 455 Bytes
41ad0fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
{
"config": {
"activation_function": "tanh",
"bias": true,
"head_type": "classification",
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2,
"LABEL_3": 3,
"LABEL_4": 4,
"LABEL_5": 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"
} |