Remove inf from stored stats (k2-fsa#1647)

yfyeung · Aug 9, 2024 · 50769ff · 50769ff
1 parent 8d68b50
commit 50769ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/icefall/utils.py b/icefall/utils.py
@@ -1159,7 +1159,8 @@ def __add__(self, other: "MetricsTracker") -> "MetricsTracker":
         for k, v in self.items():
             ans[k] = v
         for k, v in other.items():
-            ans[k] = ans[k] + v
+            if v - v == 0:
+                ans[k] = ans[k] + v
         return ans
 
     def __mul__(self, alpha: float) -> "MetricsTracker":