renamed min_len to min_length

2020-03-05 17:02:56 +01:00
parent aceb3fbaf4
commit 5b3000d933
1 changed files with 1 additions and 1 deletions
--- 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]