删除转换onnx

app.py

@@ -187,25 +187,6 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
         print_log(task_id, input2, "获得模型对象", "完成")
         yield [], log
 
-
-        if os.path.exists(onnx_path):
-            print_log(task_id, input2, "转换为ONNX模型", "跳过")
-            log += "跳过转换为ONNX模型\n"
-            yield [], log
-        else:
-            print_log(task_id, input2, "转换为ONNX模型", "开始")
-            log += "转换为ONNX模型…\n"
-            yield [], log
-            try:
-                # 使用 torch.onnx.export 进行模型转换
-                # 将列表转换为元组
-                shape_tuple = tuple(shape0)
-                torch.onnx.export(torch_model, torch.rand(shape_tuple), onnx_path, verbose=True, input_names=["data"], output_names=["output"])
-            except Exception as e:
-                print_log(task_id, input2, "转换为ONNX模型"+e, f"失败")
-                log += f"转换为ONNX模型失败: {e}\n"
-                yield [], log
-
         if os.path.exists(pt_path):
             print_log(task_id, input2, "转换为TorchScript模型", "跳过")
             log += "跳过转换为TorchScript模型\n"