xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

GPU选择debug-7

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-23 21:15:10 +08:00
parent b6f7a04d16
commit 2ebc066892
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/yolov5/train_server.py
View File

@ -217,16 +217,16 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml
best_fitness, start_epoch, epochs = smart_resume(ckpt, optimizer, ema, weights, epochs, resume)
del ckpt, csd
# DP mode
if cuda and RANK == -1 and torch.cuda.device_count() > 1:
LOGGER.warning('WARNING: DP not recommended, 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)
# # DP mode
# if cuda and RANK == -1 and torch.cuda.device_count() > 1:
# LOGGER.warning('WARNING: DP not recommended, 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
if opt.sync_bn and cuda and RANK != -1:
model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model).to(device)
LOGGER.info('Using SyncBatchNorm()')
# # SyncBatchNorm
# if opt.sync_bn and cuda and RANK != -1:
# model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model).to(device)
# LOGGER.info('Using SyncBatchNorm()')
print("Trainloader")
# Trainloader