From 74200583be92b96726de086023db2a0f45a1d401 Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Wed, 25 May 2022 17:12:47 +0800 Subject: [PATCH] minor change of the matrix_bd_utterance --- .../ASR/conv_emformer_transducer_stateless/emformer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py index 57e2c6ee2..8d989730b 100644 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py @@ -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