From 2b4ddf178eb2e3b033a6d81871750238ef1a25c4 Mon Sep 17 00:00:00 2001 From: system Date: Thu, 2 Apr 2020 11:13:53 +0000 Subject: [PATCH] Update config.json --- config.json | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/config.json b/config.json index 5b0072c..7903079 100644 --- a/config.json +++ b/config.json @@ -48,6 +48,16 @@ "prefix": " ", "pruned_heads": {}, "repetition_penalty": 1.0, + "task_specific_params": { + "summarization": { + "early_stopping": true, + "length_penalty": 2.0, + "max_length": 142, + "min_length": 56, + "no_repeat_ngram_size": 3, + "num_beams": 4 + } + }, "temperature": 1.0, "top_k": 50, "top_p": 1.0,