Skip to content

Commit 3b23406

Browse files
committed
update chunk size to 100
1 parent 12d6b73 commit 3b23406

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval_protocol/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -407,7 +407,7 @@ def parse_args(args=None):
407407
rft_parser.add_argument("--eval-auto-carveout", dest="eval_auto_carveout", action="store_true", default=True)
408408
rft_parser.add_argument("--no-eval-auto-carveout", dest="eval_auto_carveout", action="store_false")
409409
# Rollout chunking
410-
rft_parser.add_argument("--chunk-size", type=int, default=10, help="Data chunk size for rollout batching")
410+
rft_parser.add_argument("--chunk-size", type=int, default=100, help="Data chunk size for rollout batching")
411411
# Inference params
412412
rft_parser.add_argument("--temperature", type=float)
413413
rft_parser.add_argument("--top-p", type=float)

0 commit comments

Comments
 (0)