mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-15 15:13:10 +09:00
parent 3ccf18381d
commit e3ef194d43
2 changed files with 10 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
View File

Binary file not shown.

25
egs/tedlium2/ASR/conformer_ctc3/train.py
View File

@ -653,6 +653,14 @@ def compute_loss(
for i in [2,5,8,11,14] for i in [2,5,8,11,14]
] ]
ctc_loss = k2.ctc_loss(
decoding_graph=decoding_graph,
dense_fsa_vec=dense_fsa_vec,
output_beam=params.beam_size,
reduction=params.reduction,
use_double_scores=params.use_double_scores,
)
inter_ctc_loss = 0 inter_ctc_loss = 0
for fsa_vec_inter in dense_fsa_vec_inter: for fsa_vec_inter in dense_fsa_vec_inter:
inter_ctc_loss += k2.ctc_loss( inter_ctc_loss += k2.ctc_loss(
@ -663,21 +671,8 @@ def compute_loss(
use_double_scores=params.use_double_scores, use_double_scores=params.use_double_scores,
) )
ctc_loss = (1-params.interctc_weight) * ctc_loss
ctc_loss = (1-params.interctc_weight) * k2.ctc_loss( + params.interctc_weight * inter_ctc_weight
decoding_graph=decoding_graph,
dense_fsa_vec=dense_fsa_vec,
output_beam=params.beam_size,
reduction=params.reduction,
use_double_scores=params.use_double_scores,
) + params.interctc_weight * k2.ctc_loss(
decoding_graph=decoding_graph,
dense_fsa_vec=dense_fsa_vec_inter,
output_beam=params.beam_size,
reduction=params.reduction,
use_double_scores=params.use_double_scores,
)
else: else:
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(
nnet_output, nnet_output,