fix

chintu619 · May 12, 2022 · 7d52422 · 7d52422
1 parent b7cfdd9
commit 7d52422
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/espnet2/train/test_reporter.py b/test/espnet2/train/test_reporter.py
@@ -53,7 +53,7 @@ def test_register(weight1, weight2):
             desired[k] /= weight1 + weight2
 
     for k1, k2 in reporter.get_all_keys():
-        if k2 in ("time", "total_count"):
+        if k2 in ("time", "total_count", "gpu_max_cached_mem_GB", "gpu_cached_mem_GB"):
             continue
         np.testing.assert_allclose(reporter.get_value(k1, k2), desired[k2])