tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

use train_loss for sp test

Browse Source
This commit is contained in:
Wing Lian
2026-03-22 12:00:55 -04:00
parent a96733930e
commit 598c965043
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/e2e/multigpu/patched/test_sp.py
View File

@@ -94,7 +94,7 @@ class TestSequenceParallelism:
check_tensorboard( check_tensorboard(
temp_dir + "/runs", temp_dir + "/runs",
"train/loss", "train/train_loss",
threshold, threshold,
"Train Loss (%s) is too high", "Train Loss (%s) is too high",
) )