From 5b3000d9334e8f92cc9812bbc1ee7285eb26ed5e Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 5 Mar 2020 17:02:56 +0100 Subject: [PATCH] renamed min_len to min_length --- examples/summarization/bart/evaluate_cnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/summarization/bart/evaluate_cnn.py b/examples/summarization/bart/evaluate_cnn.py index f35ddcb154..e1fd8dff09 100644 --- a/examples/summarization/bart/evaluate_cnn.py +++ b/examples/summarization/bart/evaluate_cnn.py @@ -28,7 +28,7 @@ def generate_summaries(lns, out_file, batch_size=8, device=DEFAULT_DEVICE): num_beams=4, length_penalty=2.0, max_length=140, - min_len=55, + min_length=55, no_repeat_ngram_size=3, ) dec = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summaries]