diff --git a/train.py b/train.py index 6f98a64..51a0e12 100644 --- a/train.py +++ b/train.py @@ -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: