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

May resolve ultralytics#7307
ctjanuhowski · Sep 8, 2022 · 24e08b9 · 24e08b9
1 parent 4d97da1
commit 24e08b9
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: