Pārlūkot izejas kodu

Update resume.py (#4115)

modifyDataloader
Glenn Jocher GitHub pirms 3 gadiem
vecāks
revīzija
4bad914761
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      utils/aws/resume.py

+ 1
- 1
utils/aws/resume.py Parādīt failu

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


Notiek ielāde…
Atcelt
Saglabāt