Skip to content

Commit bd4250a

Browse files
committed
Fix comment clarity in maybe_evaluate function in grpo_fast.py to accurately reflect timeout logic.
1 parent da87d77 commit bd4250a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

open_instruct/grpo_fast.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2560,7 +2560,7 @@ def maybe_evaluate(
25602560
):
25612561
"""Optionally evaluate the model."""
25622562
try:
2563-
# timeout 0.01 if this is the last training step or we're not evaluating
2563+
# timeout 0.01 if this is not the last training step or we're not evaluating
25642564
# otherwise, wait to get the last evaluation generations (long timeout just in case)
25652565
timeout = 0.01 if (training_step < args.num_training_steps or args.local_eval_every < 0) else 100
25662566

0 commit comments

Comments
 (0)