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 diagnostics.py re backoff for eigs

Browse Source
This commit is contained in:
Daniel Povey 2022-12-22 23:00:44 +08:00
parent e5b047a814
commit 2e6610af5e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/diagnostics.py
View File

@ -254,7 +254,7 @@ class TensorDiagnostic(object):
"Error getting eigenvalues, trying another method."
)
eigs, _ = torch.eig(stats)
stats = eigs.abs().sqrt()
stats = eigs.norm(dim=1).sqrt()
# sqrt so it reflects data magnitude, like stddev- not variance
if stats_type in [ "rms", "stddev" ]: