From dae8f7472dee1f630d45828c186f028c05cb298f Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 25 二月 2023 19:42:25 +0800
Subject: [PATCH] Merge pull request #153 from alibaba-damo-academy/dev_gzf
---
funasr/export/export_model.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/export/export_model.py b/funasr/export/export_model.py
index e1d5fdb..933a927 100644
--- a/funasr/export/export_model.py
+++ b/funasr/export/export_model.py
@@ -58,7 +58,7 @@
if enc_size:
dummy_input = model.get_dummy_inputs(enc_size)
else:
- dummy_input = model.get_dummy_inputs_txt()
+ dummy_input = model.get_dummy_inputs()
# model_script = torch.jit.script(model)
model_script = torch.jit.trace(model, dummy_input)
@@ -111,7 +111,7 @@
dummy_input,
os.path.join(path, f'{model.model_name}.onnx'),
verbose=verbose,
- opset_version=12,
+ opset_version=14,
input_names=model.get_input_names(),
output_names=model.get_output_names(),
dynamic_axes=model.get_dynamic_axes()
--
Gitblit v1.9.1