from local

This commit is contained in:
dohe0342 2023-02-15 11:36:41 +09:00
parent 82c8cbef21
commit 20f208c4cf
2 changed files with 2 additions and 1 deletions

View File

@ -638,7 +638,6 @@ def compute_loss(
ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss
else:
'''
dense_fsa_vec = k2.DenseFsaVec(
nnet_output,
supervision_segments,
@ -652,6 +651,7 @@ def compute_loss(
reduction=params.reduction,
use_double_scores=params.use_double_scores,
)
'''
dense_fsa_vec = k2.DenseFsaVec(
nnet_output[0],
@ -706,6 +706,7 @@ def compute_loss(
)
ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss
'''
if params.att_rate > 0.0:
with torch.set_grad_enabled(is_training):