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

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-03-12 15:43:16 +08:00
parent 33c0f8f7f6
commit 949b53274c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py
View File

@ -262,7 +262,7 @@ def modified_beam_search(
for t in range(T): for t in range(T):
# fmt: off # fmt: off
current_encoder_out = encoder_out[:, t:t+1, :].unsqueeze(2) current_encoder_out = encoder_out[:, t:t+1, :].unsqueeze(2)
# current_encoder_out is of shape (1, 1, encoder_out_dim) # current_encoder_out is of shape (1, 1, 1, encoder_out_dim)
# fmt: on # fmt: on
A = list(B) A = list(B)
B = HypothesisList() B = HypothesisList()
@ -282,7 +282,7 @@ def modified_beam_search(
current_encoder_out = current_encoder_out.expand( current_encoder_out = current_encoder_out.expand(
decoder_out.size(0), 1, 1, -1 decoder_out.size(0), 1, 1, -1
) ) # (num_hyps, 1, 1, encoder_out_dim)
logits = model.joiner( logits = model.joiner(
current_encoder_out, current_encoder_out,