diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index d7c044c8d..d0835fcce 100644 Binary files a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp and b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp differ diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py index bd542cfee..078210a90 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -161,7 +161,7 @@ class Transformer(nn.Module): self.decoder_criterion = None self.group_num = group_num - self.group_layer_num = int(self.encoder_layers // self.group_num) + self.group_layer_num = int(num_encoder_layers // self.group_num) self.alpha = nn.Parameter(torch.rand(self.group_num)) self.sigmoid = nn.Sigmoid() self.layer_norm = nn.LayerNorm(d_model)