re-add scoring filtering

2020-03-10 16:53:09 +01:00
parent 9b8ee8cea0
commit 7351a8dbaf
2 changed files with 8 additions and 10 deletions
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -471,8 +471,7 @@ class BartModelIntegrationTest(unittest.TestCase):
            attention_mask=dct["attention_mask"].to(torch_device),
            num_beams=4,
            length_penalty=2.0,
-            #            max_length=max_length + 2,
-            max_length=max_length + 1,
+            max_length=max_length + 2,
            min_length=min_length + 1,
            no_repeat_ngram_size=3,
            do_sample=False,