From e433e93b807f9842df752dc1164f6117e4acb8eb Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 15 Feb 2023 15:11:45 +0900 Subject: [PATCH] from local --- egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 69632 -> 69632 bytes egs/tedlium2/ASR/conformer_ctc3/train.py | 11 +++++++++++ 2 files changed, 11 insertions(+) diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index 5e651dd6e9211fa89ff13f8c2139b9253916420f..d8ae5d3ea3c313ee362d916745bd0634f59110fa 100644 GIT binary patch delta 167 zcmZozz|ydQMKsAE%+puFQqO<^2m}}yjLP36zuYJ~TY%AT^G1Q-dR`%B28J{ih@{_U z!3n~Qnw)G546j%j7`6a$D-Z_(u{03B1S;DJ#Fapt4#a7jeJ4sXPxhR}JNdv2*2(>| zMW?UhWwe>Rz=LJ7^vr_E3nuYSkLP2Qn7m+SDp!12YI1yLUP)@vWJABe$u+ZuHk(X4 GQwabp&n}e! delta 135 zcmZozz|ydQMKsAE%+puFQqO<^2m}}yw9DTlAKEB7TY%Ad^G1Q-dfs;|3=Cpy5J~6F zf)j)pHAR8k53CFfM}fEri1UEh4~W%)_%BfTZXjL(#O<4XCrUC;zVJ?FbN`z<#>och RJW$Lvxn{P|W|L`WDgk}SBYOY< diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 847946633..da6bdc87c 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -652,6 +652,17 @@ def compute_loss( ) for i in [2,5,8,11,14] ] + + inter_ctc_loss = 0 + for fsa_vec_inter in dense_fsa_vec_inter: + inter_ctc_loss += k2.ctc_loss( + decoding_graph=decoding_graph, + dense_fsa_vec=dense_fsa_vec_inter, + output_beam=params.beam_size, + reduction=params.reduction, + use_double_scores=params.use_double_scores, + ) + ctc_loss = (1-params.interctc_weight) * k2.ctc_loss( decoding_graph=decoding_graph,