from local

This commit is contained in:
dohe0342 2023-02-14 18:53:09 +09:00
parent 7beb40420e
commit 4958225ae9
3 changed files with 20 additions and 20 deletions

View File

@ -590,31 +590,31 @@ def compute_loss(
)
dense_fsa_vec_inter = [
#k2.DenseFsaVec(
# nnet_output[1][2],
# supervision_segments,
# allow_truncate=params.subsampling_factor - 1,
#),
#k2.DenseFsaVec(
# nnet_output[1][5],
# supervision_segments,
# allow_truncate=params.subsampling_factor - 1,
#),
k2.DenseFsaVec(
nnet_output[1][2],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
),
k2.DenseFsaVec(
nnet_output[1][5],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
),
k2.DenseFsaVec(
nnet_output[1][8],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
),
#k2.DenseFsaVec(
# nnet_output[1][11],
# supervision_segments,
# allow_truncate=params.subsampling_factor - 1,
#),
#k2.DenseFsaVec(
# nnet_output[1][14],
# supervision_segments,
# allow_truncate=params.subsampling_factor - 1,
#)
k2.DenseFsaVec(
nnet_output[1][11],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
),
k2.DenseFsaVec(
nnet_output[1][14],
supervision_segments,
allow_truncate=params.subsampling_factor - 1,
)
]
ctc_loss = k2.ctc_loss(