diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp
index fd57b1891..8f51d00ad 100644
Binary files a/egs/aishell/ASR/transformer_ctc/.train.py.swp and b/egs/aishell/ASR/transformer_ctc/.train.py.swp differ
diff --git a/egs/aishell/ASR/transformer_ctc/.transformer.py.swp b/egs/aishell/ASR/transformer_ctc/.transformer.py.swp
index 16ae76b92..ab901431f 100644
Binary files a/egs/aishell/ASR/transformer_ctc/.transformer.py.swp and b/egs/aishell/ASR/transformer_ctc/.transformer.py.swp differ
diff --git a/egs/aishell/ASR/transformer_ctc/train.py b/egs/aishell/ASR/transformer_ctc/train.py
index 2b548e380..e7daf4df3 100755
--- a/egs/aishell/ASR/transformer_ctc/train.py
+++ b/egs/aishell/ASR/transformer_ctc/train.py
@@ -612,6 +612,8 @@ def run(rank, world_size, args):
         num_features=params.feature_dim,
         num_classes=num_classes,
         use_feat_batchnorm=params.use_feat_batchnorm,
+        num_encoder_layers=params.num_encoder_layers,
+        num_decoder_layers=params.num_decoder_layers,
     )
     '''
     model = Conformer(