GPU选择debug-4

2022-11-23 21:06:24 +08:00 · 2022-11-23 21:06:24 +08:00 · b527bfcb39
commit b527bfcb39
parent 3da6f21b44
1 changed files with 1 additions and 1 deletions
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@ -609,7 +609,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):

    # DDP mode
    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')
    # if LOCAL_RANK != -1:
    #     msg = 'is not compatible with YOLOv5 Multi-GPU DDP training'