diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index be0e664cc..7a65760e3 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp differ 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,