From 24b02725285c243639a0f646e058af52c256d900 Mon Sep 17 00:00:00 2001 From: "552068321@qq.com" Date: Tue, 8 Nov 2022 14:36:44 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/controller/AlgorithmController.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py index 7e54e85..75d7c38 100644 --- a/app/controller/AlgorithmController.py +++ b/app/controller/AlgorithmController.py @@ -301,9 +301,11 @@ def Export_model_RODY(params_str): params = TrainParams() params.read_from_str(params_str) exp_inputPath = params.get('exp_inputPath').value # 模型路径 + print('输入模型：', exp_inputPath) exp_device = params.get('device').value modellist = Start_Model_Export(exp_inputPath, exp_device) exp_outputPath = exp_inputPath.replace('pt', 'zip') # 压缩文件 + print('模型路径：',exp_outputPath) zipf = zipfile.ZipFile(exp_outputPath, 'w') for file in modellist: zipf.write(file, arcname=Path(file).name) # 将torchscript和onnx模型压缩