saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add current changes

Browse source
This commit is contained in:
Patrick von Platen 2020-03-06 15:14:43 +01:00
parent 333affcb81
commit 77e6775065
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_modeling_bart.py
View file

@ -515,7 +515,8 @@ class BartModelIntegrationTest(unittest.TestCase):
max_length=max_length + 2,
min_length=min_length + 1,
no_repeat_ngram_size=3,
do_sample=False
do_sample=False,
early_stopping=True
)
decoded = [
tok.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in hypotheses_batch