From feb1e56e2ebf15e169c2ec1b1e5906733429fa5b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 22 Jul 2021 17:22:11 +0200 Subject: [PATCH] Update resume.py (#4115) --- utils/aws/resume.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/aws/resume.py b/utils/aws/resume.py index 4b0d4246b594..e869834e96e7 100644 --- a/utils/aws/resume.py +++ b/utils/aws/resume.py @@ -28,7 +28,7 @@ if ddp: # multi-GPU port += 1 - cmd = f'python -m torch.distributed.launch --nproc_per_node {nd} --master_port {port} train.py --resume {last}' + cmd = f'python -m torch.distributed.run --nproc_per_node {nd} --master_port {port} train.py --resume {last}' else: # single-GPU cmd = f'python train.py --resume {last}'