Glenn Jocher
GitHub
3 years ago
1 changed files with 2 additions and 1 deletions
+ 2
- 1
train.py
View File
| if not opt.evolve: | if not opt.evolve: | ||||
| train(opt.hyp, opt, device, callbacks) | train(opt.hyp, opt, device, callbacks) | ||||
| if WORLD_SIZE > 1 and RANK == 0: | if WORLD_SIZE > 1 and RANK == 0: | ||||
| _ = LOGGER.info('Destroying process group... ', end=''), dist.destroy_process_group(), LOGGER.info('Done.') | |||||
| LOGGER.info('Destroying process group... ') | |||||
| dist.destroy_process_group() | |||||
| # Evolve hyperparameters (optional) | # Evolve hyperparameters (optional) | ||||
| else: | else: |
Loading…