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

Move PseudoNormalizeFunction to a different place.

Browse Source
This commit is contained in:
Daniel Povey 2022-06-10 14:01:13 +08:00
parent 77357cdaa8
commit 58cbc3d961
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
View File

@ -759,8 +759,8 @@ def _update_cov_stats(cov: Tensor,
x: Tensor of features/activations, of shape (num_frames, num_channels) x: Tensor of features/activations, of shape (num_frames, num_channels)
beta: The decay constant for the stats, e.g. 0.8. beta: The decay constant for the stats, e.g. 0.8.
""" """
x = PseudoNormalizeFunction.apply(x)
new_cov = torch.matmul(x.t(), x) new_cov = torch.matmul(x.t(), x)
new_cov = PseudoNormalizeFunction.apply(new_cov)
return cov * beta + new_cov * (1-beta) return cov * beta + new_cov * (1-beta)