moa-classification / head_config.json
evilfreelancer's picture
Upload 4 files
b4c94db verified
raw
history blame
554 Bytes
{
"config": {
"activation_function": "tanh",
"bias": true,
"dropout_prob": null,
"head_type": "classification",
"label2id": {
"cot": 6,
"exam": 3,
"finance": 0,
"gpt4tools": 5,
"leetcode": 2,
"medicine": 1,
"stackoverflow": 7,
"webgpt": 4
},
"layers": 2,
"num_labels": 8,
"use_pooler": false
},
"hidden_size": 768,
"model_class": "RobertaAdapterModel",
"model_name": "roberta-base",
"model_type": "roberta",
"name": "moa_classifier",
"version": "0.2.1"
}