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

Merge 8809c7a99115277a2d4a77c69714ada661ba6653 into abd9437e6d5419a497707748eb935e50976c3b7b

Browse Source
This commit is contained in:
kobenaxie 2025-06-27 11:33:58 +00:00 committed by GitHub
commit 57214969b1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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: if y_lens is not None:
# Limit the maximum number of reduced frames # 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() max_limit_len = limit_lens.max().int()
fake_limit_indexes = torch.topk( fake_limit_indexes = torch.topk(
ctc_output[:, :, blank_id], max_limit_len ctc_output[:, :, blank_id], max_limit_len
@ -97,6 +97,7 @@ class FrameReducer(nn.Module):
).scatter_(1, limit_indexes, 1) ).scatter_(1, limit_indexes, 1)
non_blank_mask = non_blank_mask | ~limit_mask non_blank_mask = non_blank_mask | ~limit_mask
non_blank_mask = non_blank_mask & ~padding_mask
out_lens = non_blank_mask.sum(dim=1) out_lens = non_blank_mask.sum(dim=1)
max_len = out_lens.max() max_len = out_lens.max()