From 503f8d521ce10d24e5fc1b62760c630843cf80c2 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 4 Mar 2022 13:08:56 +0800 Subject: [PATCH] Fix bug in diagnostics --- egs/librispeech/ASR/transducer_stateless/diagnostics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/transducer_stateless/diagnostics.py b/egs/librispeech/ASR/transducer_stateless/diagnostics.py index 2dff91805..1a2324775 100644 --- a/egs/librispeech/ASR/transducer_stateless/diagnostics.py +++ b/egs/librispeech/ASR/transducer_stateless/diagnostics.py @@ -56,7 +56,8 @@ def get_sum_abs_stats(x: Tensor, dim: int, x = (x > 0).to(dtype=torch.float) orig_numel = x.numel() sum_dims = [ d for d in range(x.ndim) if d != dim ] - x = torch.sum(x, dim=sum_dims) + if len(sum_dims) != 0: + x = torch.sum(x, dim=sum_dims) count = orig_numel // x.numel() x = x.flatten() return x, count