diff --git a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp index f500be1a5..9544784aa 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/conformer.py b/egs/librispeech/ASR/conformer_ctc2/conformer.py index 7245c1166..e3e9ce522 100644 --- a/egs/librispeech/ASR/conformer_ctc2/conformer.py +++ b/egs/librispeech/ASR/conformer_ctc2/conformer.py @@ -136,6 +136,7 @@ class Conformer(Transformer): self.group_num = group_num if self.group_num != 0: + self.learnable_alpha = learnable_alpha self.group_layer_num = int(num_encoder_layers // self.group_num) self.alpha = nn.Parameter(torch.rand(self.group_num)) self.sigmoid = nn.Sigmoid()