hnluo
2023-10-25 8a84ed6a4a6ac6958a480961dc2f1606b1ac7fb6
Merge pull request #1040 from alibaba-damo-academy/hnluo-patch-2

fix chunk size select bug
1个文件已修改
2 ■■■ 已修改文件
funasr/modules/streaming_utils/chunk_utilis.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/modules/streaming_utils/chunk_utilis.py
@@ -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)