From 25fbf7110b5fc66d592bd5e94624a0ddb60e50b8 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 19 二月 2024 21:30:11 +0800
Subject: [PATCH] aishell example
---
funasr/frontends/wav_frontend.py | 11 ++++++-----
1 files changed, 6 insertions(+), 5 deletions(-)
diff --git a/funasr/frontends/wav_frontend.py b/funasr/frontends/wav_frontend.py
index fe22335..71cf77a 100644
--- a/funasr/frontends/wav_frontend.py
+++ b/funasr/frontends/wav_frontend.py
@@ -32,6 +32,7 @@
rescale_line = line_item[3:(len(line_item) - 1)]
vars_list = list(rescale_line)
continue
+ import pdb;pdb.set_trace()
means = np.array(means_list).astype(np.float32)
vars = np.array(vars_list).astype(np.float32)
cmvn = np.array([means, vars])
@@ -399,11 +400,11 @@
return feats_pad, feats_lens, lfr_splice_frame_idxs
def forward(
- self, input: torch.Tensor, input_lengths: torch.Tensor, cache: dict = {}, **kwargs
+ self, input: torch.Tensor, input_lengths: torch.Tensor, **kwargs
):
is_final = kwargs.get("is_final", False)
- reset = kwargs.get("reset", False)
- if len(cache) == 0 or reset:
+ cache = kwargs.get("cache", {})
+ if len(cache) == 0:
self.init_cache(cache)
batch_size = input.shape[0]
@@ -448,8 +449,8 @@
feats = torch.stack(cache["lfr_splice_cache"])
feats_lengths = torch.zeros(batch_size, dtype=torch.int) + feats.shape[1]
feats, feats_lengths, _ = self.forward_lfr_cmvn(feats, feats_lengths, is_final, cache=cache)
- if is_final:
- self.init_cache(cache)
+ # if is_final:
+ # self.init_cache(cache)
return feats, feats_lengths
--
Gitblit v1.9.1