From 2b1eb92a987dc0cf98ab68fbfe4e59fb514cdbb3 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Feb 2023 02:51:33 +0900 Subject: [PATCH] from local --- egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 53248 -> 57344 bytes egs/tedlium2/ASR/conformer_ctc3/train.py | 3 ++- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index 667597d2a9c2864c47ed862be8550c442b0dd6c1..e60a017d6067d2a0b3e9be4acaeebdfd2ebf4c02 100644 GIT binary patch delta 400 zcmZozz}#?vSuDvQ%+puFQqO<^2m}}ymQ=h-X3E+q_J*I4XEL*ZE+hYBLxKB@v6BS_ z#RYjmihw{6LNSzXmJ~eBw@HCTf!B|ZfngaCgUsNcEa+g*xNLKvLnt4s0uX3!UhNkd z$jipWz%Y#&q9%5;;DTaC+g(5|Na!RGp8(=+Aoc@dUmz9*;-^3@mw|X05Z3~64G_Bl zu^15Fn|yGg_~x4n_b~>7ECGS^qWsc=_`K3w+k(WR#N1*%xPT4=SOJho&dZC>%}*)K zNsZ4=Ey_#HfvJV?;)^q@Qgt>zUo5E;m6*c7FahZ44AaI24)aa+Y~$Pfb>U}5 E0N9LP`Tzg` delta 196 zcmZoTz}&EaSuDvQ%+puFQqO<^2m}}y9+$mJ?oQh%_J*I4dor_tE~CI?LxKB@Ig`+0|cc^BzBsKwci8ew{y^*p#9~1FjERBa3J@;`;yNI%1!8v~76;<{Kx1xB XKDbbPvi)KH&HwK*GHrgbSV9K?Q!X!1 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)