From f1214f237dab515be92efeec890a66a8ab394dc5 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 13 Aug 2022 03:57:31 +0200
Subject: [PATCH] Remove DDP `destroy_process_group()` on train end (#8935)

May resolve https://github.com/ultralytics/yolov5/issues/7307
---
 train.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/train.py b/train.py
index 9f67e93..12af2a3 100644
--- 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: