from local

This commit is contained in:
dohe0342 2023-02-14 13:00:29 +09:00
parent 326a55eebc
commit 1711dfa302
2 changed files with 6 additions and 0 deletions

View File

@ -550,6 +550,12 @@ def compute_loss(
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
)
dense_fsa_vec1 = k2.DenseFsaVec(
nnet_output[-1],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
)
ctc_loss = k2.ctc_loss(
decoding_graph=decoding_graph,