From b72c4338706e46ca664e7eb3d48c0dec5808655a Mon Sep 17 00:00:00 2001 From: pikaliov Date: Tue, 21 Apr 2026 19:38:08 +0300 Subject: [PATCH] =?UTF-8?q?Reduce=20default=20batch=5Fsize=2064=E2=86=928?= =?UTF-8?q?=20(MONA=20adapters=20need=20~17GB=20at=20bs=3D8)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 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 3b536d7..1dd50f5 100644 --- a/src/training/train_gtauav.py +++ b/src/training/train_gtauav.py @@ -68,7 +68,7 @@ class TrainConfigGTAUAV: resume_from: str | None = None # path to checkpoint for resuming output_dir: str = "out/gtauav/with_text" epochs: int = 50 - batch_size: int = 64 + batch_size: int = 8 num_workers: int = 4 learning_rate: float = 1e-4 text_lr_factor: float = 0.1 # text encoder LR = learning_rate * factor @@ -517,7 +517,7 @@ def main() -> None: help="Path to seg_filter.json for excluding bad images.", ) parser.add_argument( - "--batch-size", type=int, default=64, + "--batch-size", type=int, default=8, help="Batch size.", ) parser.add_argument(