re-add scoring filtering

This commit is contained in:
Patrick von Platen committed 2020-03-11 11:06:56 +01:00
1 parent 9b8ee8cea0
commit 7351a8dbaf
2 files changed
+8 -10

No files matched your search

+1 -2
View File
@@ -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,