mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 18:12:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3 from kobenaxie/kobenaxie-patch-2

Browse Source 
Fix Blankskip bug.
This commit is contained in:
kobenaxie 2023-12-07 21:35:18 +08:00 committed by GitHub
commit 8809c7a991
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py
View File

@ -76,7 +76,7 @@ class FrameReducer(nn.Module):
if y_lens is not None:
# Limit the maximum number of reduced frames
limit_lens = T - y_lens
limit_lens = x_lens - y_lens
max_limit_len = limit_lens.max().int()
fake_limit_indexes = torch.topk(
ctc_output[:, :, blank_id], max_limit_len
@ -97,6 +97,7 @@ class FrameReducer(nn.Module):
).scatter_(1, limit_indexes, 1)
non_blank_mask = non_blank_mask | ~limit_mask
non_blank_mask = non_blank_mask & ~padding_mask
out_lens = non_blank_mask.sum(dim=1)
max_len = out_lens.max()