Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jul 22, 2021

Commit

4bad914

•

1 Parent(s): 2c073cd

Update resume.py (#4115)

Files changed (1) hide show

utils/aws/resume.py CHANGED Viewed

@@ -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}'
     else:  # single-GPU
         cmd = f'python train.py --resume {last}'

     if ddp:  # multi-GPU
         port += 1
+        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}'