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

May resolve https://github.com/ultralytics/yolov5/issues/7307
2022-08-13 03:57:31 +02:00 · 2022-08-13 03:57:31 +02:00 · f1214f237d
parent 11a76bb4f5
commit f1214f237d
1 changed files with 0 additions and 3 deletions
--- 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: