from local

This commit is contained in:
dohe0342 2023-02-16 01:46:51 +09:00
parent 31c9c40161
commit 35d6790433
2 changed files with 2 additions and 2 deletions

View File

@ -555,7 +555,7 @@ def compute_loss(
token_ids = graph_compiler.texts_to_ids(texts) token_ids = graph_compiler.texts_to_ids(texts)
decoding_graph = graph_compiler.compile(token_ids) decoding_graph = graph_compiler.compile(token_ids)
if params.interctc and not params.condition: if params.interctc:
dense_fsa_vec1 = k2.DenseFsaVec( dense_fsa_vec1 = k2.DenseFsaVec(
nnet_output[0], nnet_output[0],
supervision_segments, supervision_segments,
@ -582,7 +582,7 @@ def compute_loss(
use_double_scores=params.use_double_scores, use_double_scores=params.use_double_scores,
) )
elif params.interctc and params.condition: if params.condition and params.group_num == 0:
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(
nnet_output[0], nnet_output[0],
supervision_segments, supervision_segments,