Remove inf from stored stats (#1647)

k2-fsa · Jun 10, 2024 · 4d5c1f2 · 4d5c1f2
1 parent 130a18c
commit 4d5c1f2
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":