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

Fix bug when channel_dim < 0

Browse Source
This commit is contained in:
Daniel Povey 2022-10-13 13:40:43 +08:00
parent 9270e32a51
commit b09a1b2ae6
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -325,7 +325,10 @@ class ActivationBalancer(torch.nn.Module):
channel.
"""
with torch.no_grad():
sum_dims = [d for d in range(x.ndim) if d != self.channel_dim]
channel_dim = self.channel_dim
if channel_dim < 0:
channel_dim += x.ndim
sum_dims = [d for d in range(x.ndim) if d != channel_dim]
x_mean = torch.mean(x, dim=sum_dims).to(torch.float32)
x_abs_mean = torch.mean(x.abs(), dim=sum_dims).to(torch.float32)