Update config.json

2019-09-24 12:53:04 +00:00 · 2019-09-24 12:53:04 +00:00 · 1f0a32192b
parent 2c138db0e6
commit 1f0a32192b
1 changed files with 2 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -10,9 +10,10 @@
  "max_position_embeddings": 514,
  "num_attention_heads": 16,
  "num_hidden_layers": 24,
-  "num_labels": 2,
+  "num_labels": 3,
  "output_attentions": false,
  "output_hidden_states": false,
+  "pruned_heads": {},
  "torchscript": false,
  "type_vocab_size": 1,
  "vocab_size": 50265