Update resume.py (#4115)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-22 17:22:11 +02:00
1 parent 2c073cd207
commit 4bad914761
1 file changed
+1 -1
+1 -1
View File
@@ -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}'