Add torch DP warning (#3698)
This commit is contained in:
1 parent
b3e2f4e08d
commit
c1af67dcd4
1 file changed
+2
@@ -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
|
||||
|
||||
Reference in new issue
Block a user