diff --git a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp index e754d62b6..9a7a12f3c 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/conformer.py b/egs/tedlium2/ASR/conformer_ctc3/conformer.py index 1ba68fb65..ff113b8b0 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/conformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/conformer.py @@ -130,6 +130,13 @@ class Conformer(Transformer): ), ) + self.group_num = group_num + if self.group_num != 0: + 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) + def run_encoder( self, x: torch.Tensor,