Skip to content

Commit 83a6535

Browse files
committed
update max duration for trainer
1 parent d9562be commit 83a6535

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/2025_10_02_phase2/script.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def build_dataset_config(common: CommonComponents) -> OlmoEarthDatasetConfig:
190190

191191
def build_trainer_config(common: CommonComponents) -> TrainerConfig:
192192
"""Build the trainer config for an experiment."""
193-
MAX_DURATION = Duration.epochs(301)
193+
MAX_DURATION = Duration.steps(700000)
194194
METRICS_COLLECT_INTERVAL = 10
195195
CANCEL_CHECK_INTERVAL = 25
196196
LOAD_STRATEGY = LoadStrategy.if_available

0 commit comments

Comments
 (0)