Upload 163 files

hugging/td_fuse/td_loop.py

@@ -323,6 +323,7 @@ def _run_loop_inner(cfg: LoopConfig, shutdown_requested: list):
     # Load history
     history = load_cycle_history(cfg.output_dir)
     cycle_num = len(history) + 1
+    cycles_run_this_session = 0
     consecutive_regressions = 0
 
     # Auto-detect model if not specified
@@ -339,7 +340,7 @@ def _run_loop_inner(cfg: LoopConfig, shutdown_requested: list):
             print(f"\n  Shutdown requested — stopping before cycle {cycle_num}")
             _print_final_summary(history)
             return
-        if cfg.max_cycles > 0 and cycle_num > cfg.max_cycles:
+        if cfg.max_cycles > 0 and cycles_run_this_session >= cfg.max_cycles:
             print(f"\n  Reached max cycles ({cfg.max_cycles}). Stopping.")
             break
 
@@ -615,6 +616,7 @@ def _run_loop_inner(cfg: LoopConfig, shutdown_requested: list):
             pass
 
         # ── SHUTDOWN CHECK before starting next cycle ──
+        cycles_run_this_session += 1
         if shutdown_requested[0]:
             print(f"\n  Shutdown requested — stopping after cycle {cycle_num}")
             break