From 9b4e9cc8a0311e5243d69b73ed073e7ea441982e Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 27 三月 2024 16:05:29 +0800
Subject: [PATCH] train update
---
funasr/utils/export_utils.py | 6 ------
1 files changed, 0 insertions(+), 6 deletions(-)
diff --git a/funasr/utils/export_utils.py b/funasr/utils/export_utils.py
index 640be05..29502b1 100644
--- a/funasr/utils/export_utils.py
+++ b/funasr/utils/export_utils.py
@@ -3,10 +3,7 @@
def export_onnx(model,
data_in=None,
- type: str = "onnx",
quantize: bool = False,
- fallback_num: int = 5,
- calib_num: int = 100,
opset_version: int = 14,
**kwargs):
model_scripts = model.export(**kwargs)
@@ -19,10 +16,7 @@
m.eval()
_onnx(m,
data_in=data_in,
- type=type,
quantize=quantize,
- fallback_num=fallback_num,
- calib_num=calib_num,
opset_version=opset_version,
export_dir=export_dir,
**kwargs
--
Gitblit v1.9.1