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

GPU选择debug-4

Browse Source
This commit is contained in:
JIAKUNHAO 2022-11-23 21:06:24 +08:00
parent 3da6f21b44
commit b527bfcb39
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/yolov5/train_server.py
View File

@ -609,7 +609,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
# DDP mode # DDP mode
print('\n','opt.device:',opt.device,'\n') print('\n','opt.device:',opt.device,'\n')
device = select_device(opt.device, batch_size=opt.batch_size) device = opt.device #select_device(opt.device, batch_size=opt.batch_size)
print('\n','device:',device,'\n') print('\n','device:',device,'\n')
# if LOCAL_RANK != -1: # if LOCAL_RANK != -1:
# msg = 'is not compatible with YOLOv5 Multi-GPU DDP training' # msg = 'is not compatible with YOLOv5 Multi-GPU DDP training'