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