From 1f0a32192bf5db483071bc586659ff3a3ed5e73d Mon Sep 17 00:00:00 2001 From: system Date: Tue, 24 Sep 2019 12:53:04 +0000 Subject: [PATCH] Update config.json --- config.json | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/config.json b/config.json index b7ade3e..982f2b0 100644 --- 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