diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index 667597d2a..e60a017d6 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 514aa16ec..05c5f1d39 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -137,7 +137,6 @@ def add_model_arguments(parser: argparse.ArgumentParser) -> None: "--kernel-size", type=int, default=15, - help="Attention dimension in the conformer model.", ) @@ -881,6 +880,8 @@ def run(rank, world_size, args): dim_feedforward=params.dim_feedforward, num_encoder_layers=params.num_encoder_layers, num_decoder_layers=params.num_decoder_layers, + cnn_module_kernel=params.kernel_size, + group_num=params.group_num, ) logging.info(model)