mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update frame_reducer.py

Browse Source 
Fix mistake
This commit is contained in:
drawfish 2023-05-23 11:51:41 +08:00 committed by GitHub
parent e4b82da41c
commit 63c7402297
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
egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/frame_reducer.py
View File

@ -80,7 +80,7 @@ class FrameReducer(nn.Module):
limit_lens = T - y_lens limit_lens = T - y_lens
else: else:
# In eval mode, ensure audio that is completely silent does not make any errors # In eval mode, ensure audio that is completely silent does not make any errors
limit_lens = torch.ones_like(x_lens) limit_lens = T - torch.ones_like(x_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