瀏覽代碼
Update resume.py (#4115)
modifyDataloader
Glenn Jocher
GitHub
3 年之前
沒有發現已知的金鑰在資料庫的簽署中
GPG 金鑰 ID: 4AEE18F83AFDEB23
共有
1 個檔案被更改,包括
1 行新增 和
1 行删除
-
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}' |
|
|
|
|