[tests] switch to torchrun (#22712)

2023-04-12 08:25:45 -07:00
parent d87ef00c31
commit 1306b7d3ae
2 changed files with 4 additions and 4 deletions
--- a/tests/extended/test_trainer_ext.py
+++ b/tests/extended/test_trainer_ext.py
@@ -366,7 +366,7 @@ class TestTrainerExt(TestCasePlus):
                n_gpus_to_use = get_gpu_count()
            master_port = get_torch_dist_unique_port()
            distributed_args = f"""
-                -m torch.distributed.launch
+                -m torch.distributed.run
                --nproc_per_node={n_gpus_to_use}
                --master_port={master_port}
                {self.examples_dir_str}/pytorch/translation/run_translation.py