Merge pull request #1040 from alibaba-damo-academy/hnluo-patch-2

fix chunk size select bug
This commit is contained in:
hnluo 2023-10-25 11:01:33 +08:00 committed by GitHub
commit 8a84ed6a4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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)