Increase default epochs from 10 to 50

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-21 19:32:35 +03:00
parent 4c862b6ec7
commit 858718431b

View File

@@ -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 = 10
epochs: int = 50
batch_size: int = 64
num_workers: int = 4
learning_rate: float = 1e-4
@@ -509,7 +509,7 @@ def main() -> None:
help="Batch size.",
)
parser.add_argument(
"--epochs", type=int, default=10,
"--epochs", type=int, default=50,
help="Number of epochs.",
)
parser.add_argument(