From 3dad06f3d045e6d83a294749c1c4e60adaa34d79 Mon Sep 17 00:00:00 2001 From: JIAKUNHAO Date: Fri, 18 Nov 2022 15:32:46 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E7=BB=83=E5=8F=82=E6=95=B0=E6=A0=B8?= =?UTF-8?q?=E5=BF=83=E9=80=89=E6=8B=A9debug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/controller/AlgorithmController.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index d573149..3067d6e 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -381,7 +381,7 @@ def returnTrainParams(): 'show': True}, {"index": 2, "name": "img_size", "value": 640, "description": '训练图像大小', "default": 640, "type": "I", 'show': True}, - {"index": 3, "name": "device", "value": 'CUDA', "description": '训练核心', "default": 'CUDA', "type": "E", + {"index": 3, "name": "device", "value": f'{_kernel[0]}', "description": '训练核心', "default": f'{_kernel[0]}', "type": "E", "items": _kernel, 'show': True}, # _kernel {"index": 4, "name": "saveModDir", "value": "E:/alg_demo-master/alg_demo/app/yolov5/best.pt", "description": '保存模型路径',