训练卡住debug

app/controller/AlgorithmController.py

@@ -297,6 +297,7 @@ def error_return(id: str, data):
 # 启动训练
 @start_train_algorithm()
 def train_R0DY(params_str, id):
+    print(params_str)
     from app.yolov5.train_server import train_start
     params = TrainParams()
     params.read_from_str(params_str)
@@ -308,12 +309,12 @@ def train_R0DY(params_str, id):
     epoches = params.get('epochnum').value
     batch_size = params.get('batch_size').value
     device = params.get('device').value
-    try:
-        train_start(weights, savemodel, epoches, img_size, batch_size, device, data_list, id)
-        print("train down!")
-    except Exception as e:
-        print(repr(e))
-        error_return(id=id,data=repr(e))
+    #try:
+    train_start(weights, savemodel, epoches, img_size, batch_size, device, data_list, id)
+    print("train down!")
+    # except Exception as e:
+    #     print(repr(e))
+    #     error_return(id=id,data=repr(e))
 
 
 # 启动验证程序