diff --git a/tests/test_suites/llm/dpo-qwen2.5-math7b-1n8g-megatron_chunked_linear_ce_loss.sh b/tests/test_suites/llm/dpo-qwen2.5-math7b-1n8g-megatron_chunked_linear_ce_loss.sh index ec44a8a9600..d3a2bc7cdd0 100755 --- a/tests/test_suites/llm/dpo-qwen2.5-math7b-1n8g-megatron_chunked_linear_ce_loss.sh +++ b/tests/test_suites/llm/dpo-qwen2.5-math7b-1n8g-megatron_chunked_linear_ce_loss.sh @@ -37,7 +37,7 @@ if [[ $(jq 'to_entries | .[] | select(.key == "train/loss") | .value | keys | ma uv run tests/check_metrics.py $JSON_METRICS \ 'data["train/loss"]["1"] < 0.7' \ 'data["train/loss"]["10"] < 0.7' \ - 'data["train/accuracy"]["10"] > 0.56' + 'data["train/accuracy"]["10"] >= 0.5' # Clean up checkpoint directory after successful run to save space. rm -rf "$CKPT_DIR"