diff --git a/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp index 4eb60a7c2..dafe37dae 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc2/.train.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc2/train.py b/egs/tedlium2/ASR/conformer_ctc2/train.py index 7f376fdb5..1ee7bb4b7 100755 --- a/egs/tedlium2/ASR/conformer_ctc2/train.py +++ b/egs/tedlium2/ASR/conformer_ctc2/train.py @@ -336,9 +336,9 @@ def get_params() -> AttributeDict: # parameters for conformer "feature_dim": 80, "subsampling_factor": 4, - "encoder_dim": 512, - "nhead": 8, - "dim_feedforward": 2048, + "encoder_dim": 256, + "nhead": 4, + "dim_feedforward": 1024, "num_encoder_layers": 18, # parameters for ctc loss "beam_size": 10,