From 2ffe72af81a495f1ce237294dc377162f3cd8ec7 Mon Sep 17 00:00:00 2001
From: luomingshuang <739314837@qq.com>
Date: Mon, 21 Mar 2022 14:28:53 +0800
Subject: [PATCH] fix a rare error

---
 icefall/diagnostics.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/icefall/diagnostics.py b/icefall/diagnostics.py
index fa9b98fa0..08d1628ec 100644
--- a/icefall/diagnostics.py
+++ b/icefall/diagnostics.py
@@ -135,8 +135,13 @@ def get_diagnostics_for_dim(
             return ""
         count = sum(counts)
         stats = stats / count
-        stats, _ = torch.symeig(stats)
-        stats = stats.abs().sqrt()
+        try:
+            eigs, _ = torch.symeig(stats)
+            stats = eigs.abs().sqrt()
+        except:  # noqa
+            print("Error getting eigenvalues, trying another method.")
+            eigs, _ = torch.eigs(stats)
+            stats = eigs.abs().sqrt()
         # sqrt so it reflects data magnitude, like stddev- not variance
     elif sizes_same:
         stats = torch.stack(stats).sum(dim=0)