From 040592a9e3426271f34e3ee165611b7b72bab1e3 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 5 Oct 2022 16:22:33 +0800 Subject: [PATCH] Fix eigs call --- icefall/diagnostics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icefall/diagnostics.py b/icefall/diagnostics.py index 7807ad5e6..071d27112 100644 --- a/icefall/diagnostics.py +++ b/icefall/diagnostics.py @@ -211,7 +211,7 @@ class TensorDiagnostic(object): print( "Error getting eigenvalues, trying another method." ) - eigs = torch.linalg.eigvals(stats) + eigs, _ = torch.eig(stats) stats = eigs.abs().sqrt() # sqrt so it reflects data magnitude, like stddev- not variance