From 858718431ba2d727614b13c717e7c17459b8ad11 Mon Sep 17 00:00:00 2001 From: pikaliov Date: Tue, 21 Apr 2026 19:32:35 +0300 Subject: [PATCH] Increase default epochs from 10 to 50 Co-Authored-By: Claude Opus 4.6 (1M context) --- src/training/train_gtauav.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/training/train_gtauav.py b/src/training/train_gtauav.py index c946650..74cda12 100644 --- a/src/training/train_gtauav.py +++ b/src/training/train_gtauav.py @@ -67,7 +67,7 @@ class TrainConfigGTAUAV: # Training. resume_from: str | None = None # path to checkpoint for resuming output_dir: str = "out/gtauav/with_text" - epochs: int = 10 + epochs: int = 50 batch_size: int = 64 num_workers: int = 4 learning_rate: float = 1e-4 @@ -509,7 +509,7 @@ def main() -> None: help="Batch size.", ) parser.add_argument( - "--epochs", type=int, default=10, + "--epochs", type=int, default=50, help="Number of epochs.", ) parser.add_argument(