Massage the group

allenai · Oct 24, 2024 · c362030 · c362030
1 parent a51a63b
commit c362030
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/scripts/beaker/peteish/peteish1-eval-launch.sh b/scripts/beaker/peteish/peteish1-eval-launch.sh
@@ -7,7 +7,7 @@ NUM_NODES=16
 gantry run \
   --allow-dirty \
   --workspace ai2/OLMo-pretraining-stability \
-  --task-name peteish1 \
+  --task-name peteish1-eval \
   --description "Pete-ish 1B eval" \
   --priority high \
   --preemptible \
@@ -34,7 +34,7 @@ gantry run \
   --env-secret AWS_CREDENTIALS=PETEW_AWS_CREDENTIALS \
   --env-secret R2_ENDPOINT_URL=R2_ENDPOINT_URL \
   --env-secret WEKA_ENDPOINT_URL=WEKA_ENDPOINT_URL \
-  --env-secret WANDB_API_KEY=PETEW_WANDB_API_KEY \
+  --env-secret WANDB_API_KEY=JIACHENGL_WANDB_API_KEY \
   --shared-memory 10GiB \
   --yes \
   --timeout=-1 \

diff --git a/scripts/beaker/peteish/peteish1-eval.sh b/scripts/beaker/peteish/peteish1-eval.sh
@@ -53,6 +53,5 @@ torchrun \
       --run_name="${GANTRY_TASK_NAME}" \
       --save_interval_ephemeral=null \
       --save_overwrite \
-      --wandb.name="${GANTRY_TASK_NAME}-eval" \
-      --save_folder="/weka/oe-training-default/ai2-llm/checkpoints/OLMo-small/peteish1-eval" \
+      --wandb.group="peteish1" \
       --load_path="/weka/oe-training-default/ai2-llm/checkpoints/OLMo-small/peteish1"