from local

This commit is contained in:
dohe0342 2023-02-14 23:26:00 +09:00
parent a6c53bbdeb
commit 6de3d65cb4
2 changed files with 15 additions and 15 deletions

View File

@ -660,31 +660,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][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][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][14],
# supervision_segments,
# allow_truncate=params.subsampling_factor - 1,
#)
]
ctc_loss = k2.ctc_loss(