do 301 epochs

yawenzzzz · yawenzzzz · commit d9562be3e67e · 2025-10-22T09:44:44.000-07:00
diff --git a/olmoearth_pretrain/internal/full_eval_sweep_finetune.py b/olmoearth_pretrain/internal/full_eval_sweep_finetune.py
@@ -247,6 +247,7 @@ def _format_launch_command(
         run_name,
         cluster,
         "--launch.priority=urgent",
+        "--launch.num_gpus=1",
         "--launch.preemptible=True",
         "--launch.task_name=eval",
     ]
