Skip to content

Commit 3dfebb3

Browse files
committed
test: limit critic training to 5 steps for comparison purposes
1 parent cff7e3e commit 3dfebb3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/pipeline/rlvr_pipeline_meg_critic.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,11 @@ def run(self):
347347
actor_infer_response_timer = _Timer(window_size=5)
348348
actor_train_timer = _Timer(window_size=5)
349349

350-
for global_step in range(self.pipeline_config.max_steps):
350+
# Modified for testing: Run only 5 steps
351+
test_max_steps = min(5, self.pipeline_config.max_steps)
352+
logger.info(f"TEST MODE: Running only {test_max_steps} steps for critic comparison")
353+
354+
for global_step in range(test_max_steps):
351355
if global_step <= self.state.step:
352356
global_step += 1
353357
continue

0 commit comments

Comments
 (0)