From b79996bdc4e6a0dce3952a6abdf60668a9631378 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 6 Jul 2025 19:42:43 -0400 Subject: [PATCH] tweak loss --- tests/e2e/multigpu/test_llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py index 635e08529..906ceedea 100644 --- a/tests/e2e/multigpu/test_llama.py +++ b/tests/e2e/multigpu/test_llama.py @@ -432,7 +432,7 @@ class TestMultiGPULlama: ) check_tensorboard( - temp_dir + "/runs", "train/train_loss", 2.4, "Train Loss (%s) is too high" + temp_dir + "/runs", "train/train_loss", 2.5, "Train Loss (%s) is too high" ) @pytest.mark.parametrize(