diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py
index 1e95d06..1c5853d 100644
--- a/app/controller/AlgorithmController.py
+++ b/app/controller/AlgorithmController.py
@@ -305,7 +305,7 @@ def train_R0DY(params_str, id):
     from app.yolov5.train_server import train_start
     params = TrainParams()
     params.read_from_str(params_str)
-    print(params.get('device').default)
+    print(params.get('device').value)
     data_list = file_tool.get_file(ori_path=params.get('DatasetDir').value, type_list=params.get('CLASS_NAMES').value)
     weights = params.get('resumeModPath').value  # 初始化模型绝对路径
     img_size = params.get('img_size').value
diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py
index 717025e..271011e 100644
--- 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: