From 6d02de58ff46ae66e38c5dda44d4d254c4a56251 Mon Sep 17 00:00:00 2001 From: system Date: Thu, 26 Mar 2020 08:26:37 +0000 Subject: [PATCH] Update config.json --- config.json | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/config.json b/config.json index 37b41bc..22a3328 100644 --- a/config.json +++ b/config.json @@ -10,6 +10,7 @@ "decoder_ffn_dim": 4096, "decoder_layerdrop": 0.0, "decoder_layers": 12, + "decoder_start_token_id": 2, "do_sample": false, "dropout": 0.1, "early_stopping": true, @@ -17,9 +18,6 @@ "encoder_ffn_dim": 4096, "encoder_layerdrop": 0.0, "encoder_layers": 12, - "eos_token_ids": [ - 2 - ], "finetuning_task": null, "id2label": { "0": "LABEL_0", @@ -55,4 +53,4 @@ "torchscript": false, "use_bfloat16": false, "vocab_size": 50264 -} +} \ No newline at end of file