Update resume.py (ultralytics#4115)

fhkiel-mlaip · Jul 22, 2021 · feb1e56 · feb1e56
1 parent 6779681
commit feb1e56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}'