speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix chunk size select bug

Browse Source
This commit is contained in:
hnluo 2023-10-25 11:00:57 +08:00 committed by GitHub
parent db4a914357
commit f3330efc0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/modules/streaming_utils/chunk_utilis.py
View File

@ -61,7 +61,7 @@ class overlap_chunk():
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)