Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
utils/aws/resume.py
View File
| if ddp: # multi-GPU | if ddp: # multi-GPU | ||||
| port += 1 | 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 | else: # single-GPU | ||||
| cmd = f'python train.py --resume {last}' | cmd = f'python train.py --resume {last}' | ||||
Loading…