from local

This commit is contained in:
dohe0342 2023-02-14 13:01:08 +09:00
parent 1711dfa302
commit c2a854cf44
2 changed files with 9 additions and 2 deletions

View File

@ -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,