This commit is contained in:
speech_asr 2023-03-14 16:37:36 +08:00
parent 64b591eb6f
commit ee1b0ec605

View File

@ -430,7 +430,7 @@ class WavFrontendOnline(AbsFrontend):
else:
if is_final:
self.waveforms = waveforms if self.reserve_waveforms is None else self.reserve_waveforms
feats = torch.stack(self.lfr_splice_cache)
feats = torch.stack(self.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)
if is_final:
@ -443,4 +443,4 @@ class WavFrontendOnline(AbsFrontend):
def cache_reset(self):
self.reserve_waveforms = None
self.input_cache = None
self.lfr_splice_cache = []
self.lfr_splice_cache = []