Glenn Jocher
GitHub
3 years ago
1 changed files with 2 additions and 0 deletions
+ 2
- 0
train.py
View File
| # DP mode | # DP mode | ||||
| if cuda and RANK == -1 and torch.cuda.device_count() > 1: | 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) | model = torch.nn.DataParallel(model) | ||||
| # SyncBatchNorm | # SyncBatchNorm |
Loading…