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

Update LM test with --no-c10d

parent f66e9cb5

tests/test_binaries.py

+1 −0
Original line number Diff line number Diff line
@@ -292,6 +292,7 @@ def train_language_model(data_dir, arch): 
            '--max-epoch', '1',

            '--no-progress-bar',

            '--distributed-world-size', '1',

            '--no-c10d',

        ],

    )

    train.main(train_args)