Merge remote-tracking branch 'origin/main' into vllm_token_ids

ServiceNow · Dec 27, 2024 · 7b021f5 · 7b021f5
2 parents 7b1290b + af8df8e
commit 7b021f5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/rl_gsm8k/orchestrate_rl.py b/examples/rl_gsm8k/orchestrate_rl.py
@@ -323,6 +323,7 @@ def main(cfg: DictConfig):
     finetune_path = exp_path / "finetune"
 
     while state["iteration"] < cfg.max_iterations:
+        logger.info(f"Starting iteration {state['iteration']}")
         start_iteration = time.time()
         if os.path.exists(finetune_path / "current"):
             assistant_model_path = str(finetune_path / "current")