Fix epoch display: show 1/10 instead of 0/9

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-21 18:37:05 +03:00
parent a47dd6308e
commit f41a0f27fe
1 changed files with 1 additions and 1 deletions
--- a/src/training/train_gtauav.py
+++ b/src/training/train_gtauav.py
@@ -351,7 +351,7 @@ def train(cfg: TrainConfigGTAUAV) -> None:

        pbar = tqdm(
            train_loader,
-            desc=f"  🏋️ Epoch {epoch}/{cfg.epochs - 1}",
+            desc=f"  🏋️ Epoch {epoch + 1}/{cfg.epochs}",
            unit="batch",
            leave=False,
        )