online top-k kd

2026-02-04 09:49:35 -05:00
parent 236dad3bb7
commit a0669335e2
4 changed files with 265 additions and 2 deletions
--- a/tests/e2e/utils.py
+++ b/tests/e2e/utils.py
@@ -179,7 +179,7 @@ def check_tensorboard(
    tag: str,
    lt_val: float,
    assertion_err: str,
-    rtol: float = 0.02,
+    rtol: float = 0.05,
 ) -> None:
    """
    helper function to parse and check tensorboard logs