diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index 34a06f9..1e95d06 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -387,7 +387,7 @@ def returnTrainParams(): {"index": 2, "name": "img_size", "value": 640, "description": '训练图像大小', "default": 640, "type": "I", 'show': True}, {"index": 3, "name": "device", "value": f'{_kernel[0]}', "description": '训练核心', "default": f'{_kernel[0]}', "type": "E", - "items": _kernel, 'show': True}, # _kernel + "items": _kernel, 'show': False}, # _kernel {"index": 4, "name": "saveModDir", "value": "E:/alg_demo-master/alg_demo/app/yolov5/best.pt", "description": '保存模型路径', "default": "./app/maskrcnn/saved_model/test.pt", "type": "S", 'show': False},