from local

This commit is contained in:
dohe0342 2023-02-14 18:48:32 +09:00
parent bfa51176d0
commit b995b1d068
2 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(