From c2a854cf44939a64443673be682800f3edf5af5b Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Feb 2023 13:01:08 +0900 Subject: [PATCH] from local --- egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 57344 -> 57344 bytes egs/tedlium2/ASR/conformer_ctc3/train.py | 11 +++++++++-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index be0e664cc2abb3ce06bf81bcd4ec4a4a1cb77eb0..7a65760e3e0b44e51491d5f3a0b4dcf503d5601b 100644 GIT binary patch delta 198 zcmZoTz}#?vSv1KY%+puFQqO<^2m}}yV%T3Ndu$Y)Ex;JMd80tEKd%fE149opL^5=< zV1hKG-DXw>h87?W17Zyz{=>q+Z~%zs0&yM?I{>i|5I+NITLr{(fOs|#&jR8e=FM4& zN=$qTP>`BaoNC1|xv@!MvfaInj7F2+-?J6dRM1v1&@)!hQph&aOD;){&&e+?)|jk& V?+~LQP=N=Ffu6b6=5wh=RRC!$Fh>9Y delta 169 zcmZoTz}#?vSv1KY%+puFQqO<^2m}}yLfBs?D{T~=Ex_ovd80tEKW`{A1H(!dh@{_U z!31eWyB(|y44pt61H^hj%ml>8fJ$}%@pd2vDSragb_s|l0C6r5ivjUtCI*K6Oq;V3 om6#^qxc!eu0SZ!cic=>yG)c2S8H$rHq->o0{@&xwS!vTM0Vj+ry#N3J diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 18486646a..9ed047f55 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -557,14 +557,21 @@ def compute_loss( allow_truncate=params.subsampling_factor - 1, ) - ctc_loss = k2.ctc_loss( + ctc_loss = 0.7* k2.ctc_loss( decoding_graph=decoding_graph, - dense_fsa_vec=dense_fsa_vec, + 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( + decoding_graph=decoding_graph, + dense_fsa_vec=dense_fsa_vec2, output_beam=params.beam_size, reduction=params.reduction, use_double_scores=params.use_double_scores, ) + else: dense_fsa_vec = k2.DenseFsaVec( nnet_output,