diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index 2eca2d523..a4911e748 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 a3e50e385..06485f3b5 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -109,9 +109,14 @@ class Transformer(nn.Module): else: encoder_norm = None - self.encoder = nn.TransformerEncoder( + #self.encoder = nn.TransformerEncoder( + # encoder_layer=encoder_layer, + # num_layers=num_encoder_layers, + # norm=encoder_norm, + #) + self.encoder = TransformerEncoder( encoder_layer=encoder_layer, - num_layers=num_encoder_layers, + num_layers=encoder_layer, norm=encoder_norm, )