glenn-jocher commited on
Commit
2b3109e
1 Parent(s): 2809616

Fix DDP destruction `LOGGER.info()` (#4863)

Browse files
Files changed (1) hide show
  1. train.py +2 -1
train.py CHANGED
@@ -510,7 +510,8 @@ def main(opt, callbacks=Callbacks()):
510
  if not opt.evolve:
511
  train(opt.hyp, opt, device, callbacks)
512
  if WORLD_SIZE > 1 and RANK == 0:
513
- _ = LOGGER.info('Destroying process group... ', end=''), dist.destroy_process_group(), LOGGER.info('Done.')
 
514
 
515
  # Evolve hyperparameters (optional)
516
  else:
 
510
  if not opt.evolve:
511
  train(opt.hyp, opt, device, callbacks)
512
  if WORLD_SIZE > 1 and RANK == 0:
513
+ LOGGER.info('Destroying process group... ')
514
+ dist.destroy_process_group()
515
 
516
  # Evolve hyperparameters (optional)
517
  else: