From 6fa8ee48e117fa9c3bef450e02776e8c26b278e7 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 30 三月 2024 10:13:42 +0800
Subject: [PATCH] Dev gzf new (#1567)
---
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