fix for case of None stats

tramphero · Jul 19, 2022 · a35b28c · a35b28c
1 parent 74c14f5
commit a35b28c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/icefall/diagnostics.py b/icefall/diagnostics.py
@@ -178,6 +178,9 @@ def accumulate(self, x):
 
     def print_diagnostics(self):
         """Print diagnostics for each dimension of the tensor."""
+        if self.stats is None:
+            print(f"Warning: the stats of {self.name} is None.")
+            return
         for dim, this_dim_stats in enumerate(self.stats):
             for stats_type, stats_list in this_dim_stats.items():
                 # stats_type could be "rms", "value", "abs", "eigs", "positive".