perf: increase eval batch sizes and add max_seq_len

- humaneval/ifeval: default batch_size 64, add --max_seq_len=4096
- mmlu: batch 4 questions x 4 choices per forward, add --batch_size
- ppl: default batch_size 64
This commit is contained in:
2026-07-30 23:55:37 +08:00
parent f688cd9c5a
commit 02625739fe
4 changed files with 92 additions and 36 deletions
+5 -1
View File
@@ -509,7 +509,10 @@ def main():
help="Number of samples per problem (best-of-n scoring)",
)
parser.add_argument(
"--batch_size", type=int, default=1, help="Inference batch size"
"--batch_size", type=int, default=64, help="Inference batch size"
)
parser.add_argument(
"--max_seq_len", type=int, default=4096, help="Max sequence length for KV cache"
)
parser.add_argument(
"--limit",
@@ -542,6 +545,7 @@ def main():
model=model,
tokenizer=tokenizer,
max_batch_size=args.batch_size,
max_seq_len=args.max_seq_len,
)
results = evaluate(