From a94c4a014b069a7789c33695c0b49ac2872c906f Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 6 Jul 2025 19:25:26 -0400 Subject: [PATCH] tweak acceptable loss from changed hyperparams --- tests/e2e/multigpu/test_llama.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py index 30e7446b1..3893a8ed3 100644 --- a/tests/e2e/multigpu/test_llama.py +++ b/tests/e2e/multigpu/test_llama.py @@ -719,7 +719,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( @@ -788,7 +788,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( @@ -857,7 +857,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.skip(