Update resume.py (#4115)

2021-07-22 17:22:11 +02:00 · 2021-07-22 17:22:11 +02:00 · 4bad914761
parent 2c073cd207
commit 4bad914761
1 changed files with 1 additions and 1 deletions
--- a/utils/aws/resume.py
+++ b/utils/aws/resume.py
@ -28,7 +28,7 @@ for last in path.rglob('*/**/last.pt'):

    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}'