mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-10 13:06:29 +09:00
parent 63e2c16dc6
commit c3da411351
3 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

6
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -735,14 +735,17 @@ def compute_loss(
subsampling_factor=params.subsampling_factor, subsampling_factor=params.subsampling_factor,
token_ids=token_ids, token_ids=token_ids,
) )
logging.info('1')
# Works with a BPE model # Works with a BPE model
decoding_graph = k2.ctc_graph(token_ids, modified=False, device=device) decoding_graph = k2.ctc_graph(token_ids, modified=False, device=device)
logging.info('2')
dense_fsa_vec = k2.DenseFsaVec( dense_fsa_vec = k2.DenseFsaVec(
ctc_output, ctc_output,
supervision_segments, supervision_segments,
allow_truncate=params.subsampling_factor - 1, allow_truncate=params.subsampling_factor - 1,
) )
logging.info('3')
ctc_loss = k2.ctc_loss( ctc_loss = k2.ctc_loss(
decoding_graph=decoding_graph, decoding_graph=decoding_graph,
@ -751,6 +754,7 @@ def compute_loss(
reduction="sum", reduction="sum",
use_double_scores=params.use_double_scores, use_double_scores=params.use_double_scores,
) )
logging.info('4')
assert ctc_loss.requires_grad == is_training assert ctc_loss.requires_grad == is_training
loss += params.ctc_loss_scale * ctc_loss loss += params.ctc_loss_scale * ctc_loss

BIN
icefall/.utils.py.swp
View File

Binary file not shown.