From f41a0f27fe993b77dc3ff484100b24afbbcfdcff Mon Sep 17 00:00:00 2001
From: pikaliov <pikaliov@gmail.com>
Date: Tue, 21 Apr 2026 18:37:05 +0300
Subject: [PATCH] Fix epoch display: show 1/10 instead of 0/9

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 src/training/train_gtauav.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/training/train_gtauav.py b/src/training/train_gtauav.py
index cd1e3e6..6e5a8fc 100644
--- 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,
         )