Pikaliov/caption-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reduce default batch_size 64→8 (MONA adapters need ~17GB at bs=8)

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-21 19:38:08 +03:00
parent e3ecb09687
commit b72c433870
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/training/train_gtauav.py
View File

@@ -68,7 +68,7 @@ class TrainConfigGTAUAV:
resume_from: str | None = None # path to checkpoint for resuming resume_from: str | None = None # path to checkpoint for resuming
output_dir: str = "out/gtauav/with_text" output_dir: str = "out/gtauav/with_text"
epochs: int = 50 epochs: int = 50
batch_size: int = 64 batch_size: int = 8
num_workers: int = 4 num_workers: int = 4
learning_rate: float = 1e-4 learning_rate: float = 1e-4
text_lr_factor: float = 0.1 # text encoder LR = learning_rate * factor 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.", help="Path to seg_filter.json for excluding bad images.",
) )
parser.add_argument( parser.add_argument(
"--batch-size", type=int, default=64, "--batch-size", type=int, default=8,
help="Batch size.", help="Batch size.",
) )
parser.add_argument( parser.add_argument(