From 8b428437e9043e6012b579e9546e4e252fc3c3ee Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Wed, 23 Nov 2022 20:37:46 +0800 Subject: [PATCH] =?UTF-8?q?GPU=E9=80=89=E6=8B=A9debug-2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/yolov5/train_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index 15548ff..717025e 100644 --- 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 device = select_device(opt.device, batch_size=opt.batch_size) - print(device.type) + print('\n','device:',device,'\n') # if LOCAL_RANK != -1: # msg = 'is not compatible with YOLOv5 Multi-GPU DDP training' # assert not opt.image_weights, f'--image-weights {msg}'