SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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
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, max_length=max_length + 2,
min_length=min_length + 1, min_length=min_length + 1,
no_repeat_ngram_size=3, no_repeat_ngram_size=3,
do_sample=False do_sample=False,
early_stopping=True
) )
decoded = [ decoded = [
tok.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in hypotheses_batch tok.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in hypotheses_batch