From 9576b2cb2c280bb2a059fed501c3758fdf7ae2ba Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Sun, 24 Jul 2022 19:47:30 +0800 Subject: [PATCH] add stats for other 2 recipes --- .../ASR/conv_emformer_transducer_stateless2/train.py | 8 ++++++++ .../ASR/pruned_stateless_emformer_rnnt2/train.py | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/train.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/train.py index dfe1b6136..4706074b1 100755 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless2/train.py @@ -686,6 +686,14 @@ def compute_loss( (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. info["loss"] = loss.detach().cpu().item() info["simple_loss"] = simple_loss.detach().cpu().item() diff --git a/egs/librispeech/ASR/pruned_stateless_emformer_rnnt2/train.py b/egs/librispeech/ASR/pruned_stateless_emformer_rnnt2/train.py index cd62787fa..87fb71e1d 100755 --- a/egs/librispeech/ASR/pruned_stateless_emformer_rnnt2/train.py +++ b/egs/librispeech/ASR/pruned_stateless_emformer_rnnt2/train.py @@ -603,6 +603,14 @@ def compute_loss( (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. info["loss"] = loss.detach().cpu().item() info["simple_loss"] = simple_loss.detach().cpu().item()