mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in diagnostics

Browse Source
This commit is contained in:
Daniel Povey 2022-03-04 13:08:56 +08:00
parent 3d9ddc2016
commit 503f8d521c
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/transducer_stateless/diagnostics.py
View File

@ -56,7 +56,8 @@ def get_sum_abs_stats(x: Tensor, dim: int,
x = (x > 0).to(dtype=torch.float) x = (x > 0).to(dtype=torch.float)
orig_numel = x.numel() orig_numel = x.numel()
sum_dims = [ d for d in range(x.ndim) if d != dim ] 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() count = orig_numel // x.numel()
x = x.flatten() x = x.flatten()
return x, count return x, count