{ | |
"config": { | |
"activation_function": "tanh", | |
"bias": true, | |
"head_type": "classification", | |
"label2id": { | |
"entailment": 0, | |
"neutral": 1, | |
"contradiction": 2 | |
}, | |
"layers": 2, | |
"num_labels": 3, | |
"use_pooler": false | |
}, | |
"hidden_size": 768, | |
"model_class": "RobertaModelWithHeads", | |
"model_name": "roberta-base", | |
"model_type": "roberta", | |
"name": "glue_mnli" | |
} |