remove ddp destroy_process_group() on train end, ultralytics/yolov5#8935

modelai · Aug 16, 2022 · e28dfbc · e28dfbc
1 parent 17ee0ed
commit e28dfbc
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/README.MD b/README.MD
@@ -58,7 +58,7 @@
     | yolov5 | ✔️ | ✔️ | ✔️ | pytorch | ✔️ | local+online |
     | yolov7 | ✔️ | ✔️ | ✔️ | pytorch | ❌ | local+online |
     | mmdetection | ✔️ | ✔️ | ✔️ | pytorch | ❌ | online |
-    | detectron2 | ? | ✔️ | ✔️ | pytorch | ❌ | online |
+    | detectron2 | ✔️ | ✔️ | ✔️ | pytorch | ❌ | online |
 
     - online pretrained weights may download through network
 

diff --git a/README_zh-CN.MD b/README_zh-CN.MD
@@ -58,7 +58,7 @@
     | yolov5 | ✔️ | ✔️ | ✔️ | pytorch | ✔️ | local+online |
     | yolov7 | ✔️ | ✔️ | ✔️ | pytorch | ❌ | local+online |
     | mmdetection | ✔️ | ✔️ | ✔️ | pytorch | ❌ | online |
-    | detectron2 | ? | ✔️ | ✔️ | pytorch | ❌ | online |
+    | detectron2 | ✔️ | ✔️ | ✔️ | pytorch | ❌ | online |
 
     - online 预训练权重可能在训练时通过网络下载
 

diff --git a/det-yolov5-tmi/train.py b/det-yolov5-tmi/train.py
@@ -568,9 +568,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: