GPU选择debug-3

2022-11-23 20:52:25 +08:00
parent 8b428437e9
commit 3da6f21b44
2 changed files with 2 additions and 1 deletions
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@ -608,6 +608,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
        opt.save_dir = str(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok))

    # DDP mode
+    print('\n','opt.device:',opt.device,'\n')
    device = select_device(opt.device, batch_size=opt.batch_size)
    print('\n','device:',device,'\n')
    # if LOCAL_RANK != -1: