GPU选择debug-3

2022-11-23 20:52:25 +08:00
parent 8b428437e9
commit 3da6f21b44
2 changed files with 2 additions and 1 deletions
--- 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