dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[s2s tests] fix test_run_eval_search (#7297)

This commit is contained in:
Stas Bekman 2020-09-21 11:00:40 -07:00 committed by GitHub
parent 7a88ed6c2a
commit cbb2f75a16
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/seq2seq/test_seq2seq_examples.py
View File

@ -346,14 +346,14 @@ def test_run_eval_search(model):
task = "translation_en_to_de" if model == T5_TINY else "summarization"
testargs = f"""
run_eval_search.py
--model_name {model}
--data_dir {str(input_file_name)}
--save_dir {str(output_file_name)}
{model}
{str(input_file_name)}
{str(output_file_name)}
--score_path {score_path}
--reference_path {reference_path},
--reference_path {reference_path}
--task {task}
--search num_beams=1:2 length_penalty=0.9:1.0
""".split()
testargs.extend(["--search", "num_beams=1:2 length_penalty=0.9:1.0"])
with patch.object(sys, "argv", testargs):
with CaptureStdout() as cs: