From 2ac38adbe5f4e1374a079e032ed4b504351a207c Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 23 四月 2024 18:08:57 +0800
Subject: [PATCH] Dev gzf exp (#1647)
---
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