From f7950e6bdbf1a636bf9deeb287a81ea9abfa74a6 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Feb 2023 17:52:55 +0900 Subject: [PATCH] from local --- egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 53248 -> 53248 bytes egs/tedlium2/ASR/conformer_ctc3/train.py | 6 ++---- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index ec4acc63dd2412ae57d2ac4aa2ea1c09f7763424..7744b5032112796c49e57ede39a17810d668f008 100644 GIT binary patch delta 250 zcmZozz}&EadBb~VMuyEFm=kyy4L1i0Y_{ip#lgUEij#qXL4bk5aI#>)aaIs-%I3QP z+c_qCMw+oI0DhPv3LH8=Y$-J}Ho)G0uG delta 213 zcmWN~Jqtl$90l^qFB8yCh~ffQ5I6-QU)6ft1qA|R;6dxRo7s{e=sYu7_gWW zpMXz*GyLkDI(5dj9NTjJmZHrc+4TsMn?5djdj?U}O+-E-Q~9lvK0#~n!`d0@cdYTV z6D#Xk46f|9)1sf(iFCNTh*mIy5=7uFiT0pi00{^~2!h~+mqv7eIdq`~O-RC5;07*Y gwlUQ(=Tg&(q0UZ}Z;PBJGbxh;PCy_OxW3%R|L%t{zW@LL diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index a6884896a..5cda80235 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -308,8 +308,6 @@ def get_parser() -> argparse.ArgumentParser: "--interctc-weight", type=float, default=0.3, - help="""Number of steps that affects how rapidly the learning rate - decreases. We suggest not to change this.""", ) add_model_arguments(parser) @@ -565,13 +563,13 @@ def compute_loss( allow_truncate=params.subsampling_factor - 1, ) - ctc_loss = 0.7* k2.ctc_loss( + ctc_loss = (1-params.interctc_weight)* k2.ctc_loss( decoding_graph=decoding_graph, dense_fsa_vec=dense_fsa_vec1, output_beam=params.beam_size, reduction=params.reduction, use_double_scores=params.use_double_scores, - ) + 0.3 * k2.ctc_loss( + ) + params.interctc_weight * k2.ctc_loss( decoding_graph=decoding_graph, dense_fsa_vec=dense_fsa_vec2, output_beam=params.beam_size,