From a6b5fa5a88050e2a656bfb6f85057f57f05f7a51 Mon Sep 17 00:00:00 2001 From: system Date: Thu, 7 May 2020 02:25:21 +0000 Subject: [PATCH] Update config.json --- config.json | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/config.json b/config.json index 9c48b55..12eb8e7 100644 --- a/config.json +++ b/config.json @@ -8,6 +8,11 @@ "MarianMTModel" ], "attention_dropout": 0.0, + "bad_words_ids": [ + [ + 59513 + ] + ], "bos_token_id": 0, "classif_dropout": 0.0, "d_model": 512, @@ -34,6 +39,7 @@ "LABEL_1": 1, "LABEL_2": 2 }, + "max_length": 512, "max_position_embeddings": 512, "model_type": "marian", "normalize_before": false,