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(