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)
