Unverified 提交 4bad9147 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update resume.py (#4115)

上级 2c073cd2
......@@ -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}'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论