From d23db9512fc9a880f1a516910e6186df43c4d6fd Mon Sep 17 00:00:00 2001 From: system Date: Tue, 24 Mar 2020 20:17:24 +0000 Subject: [PATCH] Update config.json --- config.json | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index 18e15e0..37b41bc 100644 --- a/config.json +++ b/config.json @@ -12,7 +12,7 @@ "decoder_layers": 12, "do_sample": false, "dropout": 0.1, - "early_stopping": false, + "early_stopping": true, "encoder_attention_heads": 16, "encoder_ffn_dim": 4096, "encoder_layerdrop": 0.0, @@ -35,9 +35,9 @@ "LABEL_2": 2 }, "length_penalty": 2.0, - "max_length": 20, + "max_length": 142, "max_position_embeddings": 1024, - "min_length": 0, + "min_length": 56, "model_type": "bart", "no_repeat_ngram_size": 3, "num_beams": 4,