Skip to content

GitLab

Commit 8bd8ec8f authored by Myle Ott's avatar Myle Ott
Browse files

Update LM test with --no-c10d

parent f66e9cb5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_binaries.py
View file @ 8bd8ec8f
...@@ -292,6 +292,7 @@ def train_language_model(data_dir, arch): ...@@ -292,6 +292,7 @@ def train_language_model(data_dir, arch):
'--max-epoch', '1', '--max-epoch', '1',
'--no-progress-bar', '--no-progress-bar',
'--distributed-world-size', '1', '--distributed-world-size', '1',
'--no-c10d',
], ],
) )
train.main(train_args) train.main(train_args)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment