gpt2_sentiment_sst-2_pfeiffer / head_config.json
calpt's picture
Add adapter gpt2_sentiment_sst-2_pfeiffer version 1
c961366 verified
raw
history blame
407 Bytes
{
"config": {
"activation_function": null,
"bias": false,
"dropout_prob": null,
"head_type": "classification",
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"layers": 1,
"num_labels": 2,
"use_pooler": false
},
"hidden_size": 768,
"model_class": "GPT2AdapterModel",
"model_name": "gpt2",
"model_type": "gpt2",
"name": "sst2",
"version": "0.2.0"
}