mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

add stats for other 2 recipes

Browse Source
This commit is contained in:
yaozengwei 2022-07-24 19:47:30 +08:00
parent 6455fa40b8
commit 9576b2cb2c
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/conv_emformer_transducer_stateless2/train.py
View File

@ -686,6 +686,14 @@ def compute_loss(
(feature_lens // params.subsampling_factor).sum().item() (feature_lens // params.subsampling_factor).sum().item()
) )
info["utterances"] = feature.size(0)
# `utt_duration` and `utt_pad_proportion` would be normalized by `utterances` # noqa
info["utt_duration"] = feature_lens.sum().item()
# padding proportion of each utterance
info["utt_pad_proportion"] = (
((feature.size(1) - feature_lens) / feature.size(1)).sum().item()
)
# Note: We use reduction=sum while computing the loss. # Note: We use reduction=sum while computing the loss.
info["loss"] = loss.detach().cpu().item() info["loss"] = loss.detach().cpu().item()
info["simple_loss"] = simple_loss.detach().cpu().item() info["simple_loss"] = simple_loss.detach().cpu().item()

8
egs/librispeech/ASR/pruned_stateless_emformer_rnnt2/train.py
View File

@ -603,6 +603,14 @@ def compute_loss(
(feature_lens // params.subsampling_factor).sum().item() (feature_lens // params.subsampling_factor).sum().item()
) )
info["utterances"] = feature.size(0)
# `utt_duration` and `utt_pad_proportion` would be normalized by `utterances` # noqa
info["utt_duration"] = feature_lens.sum().item()
# padding proportion of each utterance
info["utt_pad_proportion"] = (
((feature.size(1) - feature_lens) / feature.size(1)).sum().item()
)
# Note: We use reduction=sum while computing the loss. # Note: We use reduction=sum while computing the loss.
info["loss"] = loss.detach().cpu().item() info["loss"] = loss.detach().cpu().item()
info["simple_loss"] = simple_loss.detach().cpu().item() info["simple_loss"] = simple_loss.detach().cpu().item()