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

filter inf loss in ctc_loss

Browse Source
This commit is contained in:
yaozengwei 2023-02-15 22:22:21 +08:00
parent 8e893576af
commit 20207f0e4e
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
egs/librispeech/ASR/pruned_transducer_stateless4_ctc/train.py
View File

@ -721,12 +721,25 @@ def compute_loss(
dense_fsa_vec=dense_fsa_vec,
output_beam=params.beam_size,
delay_penalty=params.ctc_delay_penalty if warmup >= 1.0 else 0.0,
reduction="sum",
reduction="none",
use_double_scores=params.use_double_scores,
)
assert ctc_loss.requires_grad == is_training
loss += params.ctc_loss_scale * ctc_loss
ctc_loss_is_finite = torch.isfinite(ctc_loss)
if not torch.all(ctc_loss_is_finite):
logging.info("Not all losses are finite!\n" f"ctc_loss: {ctc_loss}")
ctc_loss = ctc_loss[ctc_loss_is_finite]
# If either all simple_loss or pruned_loss is inf or nan,
# we stop the training process by raising an exception
if torch.all(~ctc_loss_is_finite):
raise ValueError(
"There are too many utterances in this batch "
"leading to inf or nan losses."
)
ctc_loss = ctc_loss.sum()
assert ctc_loss.requires_grad == is_training
loss += params.ctc_loss_scale * ctc_loss
assert loss.requires_grad == is_training
info = MetricsTracker()