From 7795224305051600c0fdd4c056f3dd6d12aa2ee7 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Sun, 4 Jan 2026 15:35:49 -0800 Subject: [PATCH] fix: grad norm check for automodel gpt oss nightly Signed-off-by: Hemil Desai --- .../test_suites/llm/sft-gpt-oss-20b-1n8g-fsdp8ep8-automodel.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_suites/llm/sft-gpt-oss-20b-1n8g-fsdp8ep8-automodel.sh b/tests/test_suites/llm/sft-gpt-oss-20b-1n8g-fsdp8ep8-automodel.sh index 24b5977f7e..9f48544b42 100755 --- a/tests/test_suites/llm/sft-gpt-oss-20b-1n8g-fsdp8ep8-automodel.sh +++ b/tests/test_suites/llm/sft-gpt-oss-20b-1n8g-fsdp8ep8-automodel.sh @@ -36,5 +36,6 @@ if [[ $(jq 'to_entries | .[] | select(.key == "train/loss") | .value | keys | ma uv run tests/check_metrics.py $JSON_METRICS \ 'data["train/loss"]["1"] < 7.0' \ 'data["train/loss"]["50"] < 0.4' \ - 'data["train/grad_norm"]["50"] < 2.5' + 'data["train/grad_norm"]["50"] < 17.5' \ + 'data["train/grad_norm"]["50"] > 10.0' fi