Fix dp size for evaluation script

This commit is contained in:
Wei Guo 2025-02-25 20:55:08 +08:00
parent 8fc76fadf0
commit 82c9b08e4b
1 changed files with 1 additions and 1 deletions

View File

@ -37,7 +37,7 @@ python3 -u math_eval.py \
--end -1 \
--use_vllm \
--max_tokens_per_call=$MAX_GEN_TOKENS \
--data_parallel_size 8 \
--data_parallel_size 2 \
--save_outputs \
# --overwrite \