From 207357692f9814dba80bdca932b5d55aa4df60f2 Mon Sep 17 00:00:00 2001 From: system Date: Thu, 13 Feb 2020 21:46:16 +0000 Subject: [PATCH] Update config.json --- config.json | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index 9b15f3b..90bf203 100644 --- a/config.json +++ b/config.json @@ -1,6 +1,10 @@ { "activation_dropout": 0.0, - "architectures": null, + "architectures": [ + "BartModel", + "BartForMaskedLM", + "BartForSequenceClassification" + ], "attention_dropout": 0.0, "bos_token_id": 0, "classif_dropout": 0.0, @@ -33,11 +37,11 @@ "model_type": "bart", "num_beams": 1, "num_hidden_layers": 12, - "num_labels": 3, + "num_labels": 2, "num_return_sequences": 1, "output_attentions": false, "output_hidden_states": false, - "output_past": true, + "output_past": false, "pad_token_id": 1, "pruned_heads": {}, "repetition_penalty": 1.0,