From 2e36e738ca39afc8d02f3d11013bd12f937cc874 Mon Sep 17 00:00:00 2001 From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com> Date: 星期三, 08 十一月 2023 09:22:06 +0800 Subject: [PATCH] fix bug for h5 hotwords (#1067) --- funasr/modules/streaming_utils/chunk_utilis.py | 4 ++-- 1 files changed, 2 insertions(+), 2 deletions(-) diff --git a/funasr/modules/streaming_utils/chunk_utilis.py b/funasr/modules/streaming_utils/chunk_utilis.py index ea37c68..73dbaf4 100644 --- a/funasr/modules/streaming_utils/chunk_utilis.py +++ b/funasr/modules/streaming_utils/chunk_utilis.py @@ -11,7 +11,7 @@ class overlap_chunk(): """ - author: Speech Lab, Alibaba Group, China + Author: Speech Lab of DAMO Academy, Alibaba Group San-m: Memory equipped self-attention for end-to-end speech recognition https://arxiv.org/abs/2006.01713 @@ -61,7 +61,7 @@ chunk_num = len(self.chunk_size) ind = 0 if training and chunk_num > 1: - ind = torch.randint(0, chunk_num-1, ()).cpu().item() + ind = torch.randint(0, chunk_num, ()).cpu().item() if not training and decoding_ind is not None: ind = int(decoding_ind) -- Gitblit v1.9.1