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