use train_loss for sp test

2026-03-22 12:00:55 -04:00
parent a96733930e
commit 598c965043
1 changed files with 1 additions and 1 deletions
--- a/tests/e2e/multigpu/patched/test_sp.py
+++ b/tests/e2e/multigpu/patched/test_sp.py
@@ -94,7 +94,7 @@ class TestSequenceParallelism:

        check_tensorboard(
            temp_dir + "/runs",
-            "train/loss",
+            "train/train_loss",
            threshold,
            "Train Loss (%s) is too high",
        )