From 0abba9e7a2eb849164459ee5ec22d7b2da28d9c5 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 12 Mar 2022 11:20:44 +0800 Subject: [PATCH] Fix self.post-scale-mha --- egs/librispeech/ASR/transducer_stateless/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)))