Remove DDP destroy_process_group() on train end (#8935)

May resolve #7307
ultralytics · Aug 13, 2022 · f1214f2 · f1214f2
1 parent 11a76bb
commit f1214f2
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -527,9 +527,6 @@ def main(opt, callbacks=Callbacks()):
     # Train
     if not opt.evolve:
         train(opt.hyp, opt, device, callbacks)
-        if WORLD_SIZE > 1 and RANK == 0:
-            LOGGER.info('Destroying process group... ')
-            dist.destroy_process_group()
 
     # Evolve hyperparameters (optional)
     else: