diff --git a/egs/librispeech/ASR/transducer_stateless/conformer.py b/egs/librispeech/ASR/transducer_stateless/conformer.py index 963cb2cd9..3f9becded 100644 --- a/egs/librispeech/ASR/transducer_stateless/conformer.py +++ b/egs/librispeech/ASR/transducer_stateless/conformer.py @@ -231,7 +231,7 @@ class ConformerEncoderLayer(nn.Module): attn_mask=src_mask, key_padding_mask=src_key_padding_mask, )[0] - src = residual + post_scale_mha(self.dropout(src_att)) + src = residual + self.post_scale_mha(self.dropout(src_att)) # convolution module src = src + self.dropout(self.conv_module(self.scale_conv(src)))