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

[deepspeed tests] fix summarization (#15149)

Browse Source
This commit is contained in:
Stas Bekman
2022-01-13 13:48:51 -08:00
committed by GitHub
parent 6e058e84fd
commit 1eb40338ac
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/deepspeed/test_model_zoo.py
View File

@@ -157,6 +157,7 @@ def make_task_cmds():
--train_file {data_dir_xsum}/sample.json --train_file {data_dir_xsum}/sample.json
--max_source_length 12 --max_source_length 12
--max_target_length 12 --max_target_length 12
--lang en
""", """,
clm=f""" clm=f"""
{scripts_dir}/language-modeling/run_clm.py {scripts_dir}/language-modeling/run_clm.py