Reduce default batch_size 64→8 (MONA adapters need ~17GB at bs=8)
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user