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

minor change of the matrix_bd_utterance

Browse Source
This commit is contained in:
yaozengwei 2022-05-25 17:12:47 +08:00
parent 435a01dbdf
commit 74200583be
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
View File

@ -730,7 +730,9 @@ class EmformerAttention(nn.Module):
matrix_bd_utterance = self._rel_shift(matrix_bd_utterance)
# (B, nhead, U, U + right_context_length) for training and validation mode; # noqa
# (B, nhead, U, tot_left_length + U + right_context_length) for inference mode. # noqa
matrix_bd_utterance = matrix_bd_utterance.view(B * self.nhead, U, -1)
matrix_bd_utterance = matrix_bd_utterance.contiguous().view(
B * self.nhead, U, -1
)
matrix_bd = torch.zeros_like(matrix_ac)
if left_context_key is not None and left_context_val is not None:
# inference mode