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

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-20 13:12:48 +02:00
1 parent 2809616345
commit 2b3109eeb0
1 file changed
+2 -1
+2 -1
View File
@@ -510,7 +510,8 @@ def main(opt, callbacks=Callbacks()):
if not opt.evolve:
train(opt.hyp, opt, device, callbacks)
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)
else: