make sure to use tensorboard to capture loss for checks

2025-01-08 22:30:13 -05:00
parent 42d3e36a6f
commit 9db5072407
1 changed files with 1 additions and 0 deletions
--- a/tests/e2e/integrations/test_kd.py
+++ b/tests/e2e/integrations/test_kd.py
@@ -64,6 +64,7 @@ def min_cfg(temp_dir):
        "max_steps": 5,
        "output_dir": temp_dir,
        "save_safetensors": True,
+        "use_tensorboard": True,
    }