diff --git a/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp b/egs/librispeech/ASR/conformer_ctc2/.conformer.py.swp new file mode 100644 index 000000000..334e164f0 Binary files /dev/null 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 ed776be53..705d48c89 100644 --- a/egs/librispeech/ASR/conformer_ctc2/conformer.py +++ b/egs/librispeech/ASR/conformer_ctc2/conformer.py @@ -105,7 +105,7 @@ class Conformer(Transformer): self.group_num = group_num if self.group_num != 0: - 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)