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

fixes

Browse Source
This commit is contained in:
Dan Saunders
2025-08-20 17:41:33 +00:00
parent 1b7b67d06e
commit 7eba3795fe
3 changed files with 39 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/e2e/test_streaming.py
View File

@@ -79,7 +79,7 @@ class TestStreamingDatasets:
check_tensorboard(
temp_dir + "/runs",
"train/train_loss",
25.0, # Loss should be reasonable for a smoke test (higher threshold for streaming)
2.5, # Loss should be reasonable for a smoke test (higher threshold for streaming)
"Train Loss (%s) is too high",
)
@@ -151,13 +151,13 @@ class TestStreamingDatasets:
check_tensorboard(
temp_dir + "/runs",
"train/train_loss",
25.0,
2.5,
"Train Loss (%s) is too high",
)
check_tensorboard(
temp_dir + "/runs",
"eval/eval_loss",
25.0,
2.5,
"Eval Loss (%s) is too high",
)
@@ -256,6 +256,6 @@ class TestStreamingDatasets:
check_tensorboard(
temp_dir + "/runs",
"train/train_loss",
25.0,
2.5,
"Train Loss (%s) is too high",
)