Update config.json
This commit is contained in:
parent
7c2a3205fc
commit
83205b1a91
|
@ -3,7 +3,8 @@
|
||||||
"RobertaForMaskedLM"
|
"RobertaForMaskedLM"
|
||||||
],
|
],
|
||||||
"attention_probs_dropout_prob": 0.1,
|
"attention_probs_dropout_prob": 0.1,
|
||||||
"finetuning_task": null,
|
"bos_token_id": 0,
|
||||||
|
"eos_token_id": 2,
|
||||||
"hidden_act": "gelu",
|
"hidden_act": "gelu",
|
||||||
"hidden_dropout_prob": 0.1,
|
"hidden_dropout_prob": 0.1,
|
||||||
"hidden_size": 768,
|
"hidden_size": 768,
|
||||||
|
@ -11,12 +12,10 @@
|
||||||
"intermediate_size": 3072,
|
"intermediate_size": 3072,
|
||||||
"layer_norm_eps": 1e-05,
|
"layer_norm_eps": 1e-05,
|
||||||
"max_position_embeddings": 514,
|
"max_position_embeddings": 514,
|
||||||
|
"model_type": "roberta",
|
||||||
"num_attention_heads": 12,
|
"num_attention_heads": 12,
|
||||||
"num_hidden_layers": 6,
|
"num_hidden_layers": 6,
|
||||||
"num_labels": 2,
|
"pad_token_id": 1,
|
||||||
"output_attentions": false,
|
|
||||||
"output_hidden_states": false,
|
|
||||||
"torchscript": false,
|
|
||||||
"type_vocab_size": 1,
|
"type_vocab_size": 1,
|
||||||
"vocab_size": 50265
|
"vocab_size": 50265
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue