From 9a7ea0f10e936277674f7e4cd17d4e9616706bff Mon Sep 17 00:00:00 2001
From: haoneng.lhn <haoneng.lhn@alibaba-inc.com>
Date: 星期二, 19 九月 2023 14:51:07 +0800
Subject: [PATCH] fix large_dataset epoch shuffle bug

---
 funasr/runtime/onnxruntime/src/paraformer-online.h |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/runtime/onnxruntime/src/paraformer-online.h b/funasr/runtime/onnxruntime/src/paraformer-online.h
index d0265f2..86ee363 100644
--- a/funasr/runtime/onnxruntime/src/paraformer-online.h
+++ b/funasr/runtime/onnxruntime/src/paraformer-online.h
@@ -101,7 +101,7 @@
         void AddOverlapChunk(std::vector<std::vector<float>> &wav_feats, bool input_finished);
         
         string ForwardChunk(std::vector<std::vector<float>> &wav_feats, bool input_finished);
-        string Forward(float* din, int len, bool input_finished);
+        string Forward(float* din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb={{0.0}});
         string Rescoring();
         // 2pass
         std::string online_res;

--
Gitblit v1.9.1