From fc40bfea8222400ffdcb437d0d4708053a619cb2 Mon Sep 17 00:00:00 2001
From: "LIyong.Guo" <839019390@qq.com>
Date: Thu, 31 Mar 2022 10:43:46 +0800
Subject: [PATCH] fix typo of torch.eig (#281)

Co-authored-by: glynpu <glynwpu@qq.com>
---
 icefall/diagnostics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/icefall/diagnostics.py b/icefall/diagnostics.py
index 08d1628ec..ce4ac1464 100644
--- a/icefall/diagnostics.py
+++ b/icefall/diagnostics.py
@@ -140,7 +140,7 @@ def get_diagnostics_for_dim(
             stats = eigs.abs().sqrt()
         except:  # noqa
             print("Error getting eigenvalues, trying another method.")
-            eigs, _ = torch.eigs(stats)
+            eigs = torch.linalg.eigvals(stats)
             stats = eigs.abs().sqrt()
         # sqrt so it reflects data magnitude, like stddev- not variance
     elif sizes_same: