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