roberta-large-mnli/config.json

33 lines
688 B
JSON
Raw Normal View History

2019-08-05 01:03:17 +00:00
{
2020-04-24 16:05:44 +00:00
"_num_labels": 3,
2020-01-31 23:00:29 +00:00
"architectures": [
"RobertaForSequenceClassification"
],
2019-08-05 01:03:17 +00:00
"attention_probs_dropout_prob": 0.1,
2020-04-24 16:05:44 +00:00
"bos_token_id": 0,
"eos_token_id": 2,
2019-08-05 01:03:17 +00:00
"hidden_act": "gelu",
"hidden_dropout_prob": 0.1,
"hidden_size": 1024,
2020-04-24 16:05:44 +00:00
"id2label": {
2020-06-17 14:42:29 +00:00
"0": "CONTRADICTION",
"1": "NEUTRAL",
"2": "ENTAILMENT"
2020-04-24 16:05:44 +00:00
},
2019-08-05 01:03:17 +00:00
"initializer_range": 0.02,
"intermediate_size": 4096,
2020-04-24 16:05:44 +00:00
"label2id": {
2020-06-17 14:42:29 +00:00
"CONTRADICTION": 0,
"NEUTRAL": 1,
"ENTAILMENT": 2
2020-04-24 16:05:44 +00:00
},
2019-08-31 00:26:50 +00:00
"layer_norm_eps": 1e-05,
2019-09-26 17:03:17 +00:00
"max_position_embeddings": 514,
2020-04-24 16:05:44 +00:00
"model_type": "roberta",
2019-08-05 01:03:17 +00:00
"num_attention_heads": 16,
"num_hidden_layers": 24,
2020-04-24 16:05:44 +00:00
"pad_token_id": 1,
2019-10-01 23:41:03 +00:00
"type_vocab_size": 1,
2019-08-05 01:03:17 +00:00
"vocab_size": 50265
}