Update config.json
This commit is contained in:
parent
2c138db0e6
commit
1f0a32192b
|
@ -10,9 +10,10 @@
|
||||||
"max_position_embeddings": 514,
|
"max_position_embeddings": 514,
|
||||||
"num_attention_heads": 16,
|
"num_attention_heads": 16,
|
||||||
"num_hidden_layers": 24,
|
"num_hidden_layers": 24,
|
||||||
"num_labels": 2,
|
"num_labels": 3,
|
||||||
"output_attentions": false,
|
"output_attentions": false,
|
||||||
"output_hidden_states": false,
|
"output_hidden_states": false,
|
||||||
|
"pruned_heads": {},
|
||||||
"torchscript": false,
|
"torchscript": false,
|
||||||
"type_vocab_size": 1,
|
"type_vocab_size": 1,
|
||||||
"vocab_size": 50265
|
"vocab_size": 50265
|
||||||
|
|
Loading…
Reference in New Issue