from local

This commit is contained in:
dohe0342 2023-02-17 11:02:55 +09:00
parent a5601c5456
commit 9777621c72
2 changed files with 3 additions and 0 deletions

View File

@ -675,6 +675,9 @@ def compute_loss(
ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss ctc_loss = (1-params.interctc_weight) * ctc_loss + params.interctc_weight * inter_ctc_loss
if not params.interctc and not params.condition: if not params.interctc and not params.condition:
if type(nnet_output) == tuple:
nnet_output = nnet_output[0]
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(
nnet_output, nnet_output,
supervision_segments, supervision_segments,