mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-14 18:02:48 +09:00
parent 1bd60745ee
commit f6b4c61d99
2 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
View File

Binary file not shown.

26
egs/tedlium2/ASR/conformer_ctc3/train.py
View File

@ -589,8 +589,8 @@ def compute_loss(
allow_truncate=params.subsampling_factor - 1, allow_truncate=params.subsampling_factor - 1,
) )
dense_fsa_vec2 = [ dense_fsa_vec_inter = [
k2.DenseFsaVec( k2.DenseFsaVec(
nnet_output[1][2], nnet_output[1][2],
supervision_segments, supervision_segments,
allow_truncate=params.subsampling_factor - 1, allow_truncate=params.subsampling_factor - 1,
@ -616,19 +616,25 @@ def compute_loss(
allow_truncate=params.subsampling_factor - 1, allow_truncate=params.subsampling_factor - 1,
) )
] ]
ctc_loss = (1-params.interctc_weight)* k2.ctc_loss(
ctc_loss = k2.ctc_loss(
decoding_graph=decoding_graph, decoding_graph=decoding_graph,
dense_fsa_vec=dense_fsa_vec1, dense_fsa_vec=dense_fsa_vec1,
output_beam=params.beam_size, output_beam=params.beam_size,
reduction=params.reduction, reduction=params.reduction,
use_double_scores=params.use_double_scores, use_double_scores=params.use_double_scores,
) + params.interctc_weight * k2.ctc_loss( )
decoding_graph=decoding_graph,
dense_fsa_vec=dense_fsa_vec2, inter_ctc_loss = 0
output_beam=params.beam_size, for fsa in dense_fsa_vec_inter:
reduction=params.reduction, inter_ctc_loss += k2.ctc_loss(
use_double_scores=params.use_double_scores, decoding_graph=decoding_graph,
) dense_fsa_vec=fsa,
output_beam=params.beam_size,
reduction=params.reduction,
use_double_scores=params.use_double_scores,
)
else: else:
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(