From e9f1dcc081fef405c809f58c491a80821f58026c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 07 二月 2023 17:52:29 +0800
Subject: [PATCH] export model
---
funasr/export/test_onnx.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/export/test_onnx.py b/funasr/export/test_onnx.py
index d1b4209..91b128e 100644
--- a/funasr/export/test_onnx.py
+++ b/funasr/export/test_onnx.py
@@ -9,7 +9,7 @@
output_name = [nd.name for nd in sess.get_outputs()]
def _get_feed_dict(feats_length):
- return {'speech': np.zeros((1, feats_length, 560), dtype=np.float32), 'speech_lengths': np.array([feats_length,], dtype=np.int64)}
+ return {'speech': np.zeros((1, feats_length, 560), dtype=np.float32), 'speech_lengths': np.array([feats_length,], dtype=np.int32)}
def _run(feed_dict):
output = sess.run(output_name, input_feed=feed_dict)
--
Gitblit v1.9.1