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,
         )