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

fhkiel-mlaip · Sep 20, 2021 · 0cc5a62 · 0cc5a62
1 parent a3cc475
commit 0cc5a62
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: