From 2db3dff8193a6dd644cd21a8b6bba5bca36a66e9 Mon Sep 17 00:00:00 2001
From: pikaliov <pikaliov@gmail.com>
Date: Tue, 21 Apr 2026 19:40:00 +0300
Subject: [PATCH] Set default epochs to 20

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 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 1dd50f5..f594ca6 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 = 50
+    epochs: int = 20
     batch_size: int = 8
     num_workers: int = 4
     learning_rate: float = 1e-4
@@ -521,7 +521,7 @@ def main() -> None:
         help="Batch size.",
     )
     parser.add_argument(
-        "--epochs", type=int, default=50,
+        "--epochs", type=int, default=20,
         help="Number of epochs.",
     )
     parser.add_argument(