Add torch DP warning (#3698)

ultralytics · Jun 19, 2021 · c1af67d · c1af67d
1 parent b3e2f4e
commit c1af67d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -198,6 +198,8 @@ def train(hyp,  # path/to/hyp.yaml or hyp dictionary
 
     # DP mode
     if cuda and RANK == -1 and torch.cuda.device_count() > 1:
+        logging.warning('DP not recommended, instead use torch.distributed.run for best DDP Multi-GPU results.\n'
+                        'See Multi-GPU Tutorial at https://github.com/ultralytics/yolov5/issues/475 to get started.')
         model = torch.nn.DataParallel(model)
 
     # SyncBatchNorm