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": '保存模型路径',