Upload code/run_polyedit_multiobjective_local.sh with huggingface_hub

code/run_polyedit_multiobjective_local.sh

@@ -0,0 +1,54 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+repo=/home/sjkim/POLYEDIT
+run_root=$repo/project/experiments/local_runs/multiobjective_20260721
+ckpt_root=/tmp/polyedit_multiobjective_ckpts_20260721
+hf_root=/tmp/polyedit_multiobjective_hf_20260721
+wandb_root=/tmp/polyedit_multiobjective_wandb_20260721
+polybert=$hf_root/polybert
+python_bin=/home/sjkim/anaconda3/envs/main/bin/python
+revision=7bf9ed32ac54dea5bc163cf90100728b49341750
+
+cleanup() {
+  rm -rf "$hf_root" "$wandb_root"
+}
+trap cleanup EXIT
+
+if [[ -z ${WANDB_API_KEY:-} ]]; then
+  read -rsp "W&B API key: " WANDB_API_KEY
+  export WANDB_API_KEY
+  echo
+fi
+mkdir -p "$run_root/results" "$run_root/logs" "$ckpt_root" "$hf_root" "$wandb_root"
+export PYTHONPATH=$repo
+export HF_HOME=$hf_root/cache
+export HF_HUB_CACHE=$hf_root/cache/hub
+export WANDB_DIR=$wandb_root
+export TOKENIZERS_PARALLELISM=false
+
+/home/sjkim/anaconda3/envs/main/bin/hf download HAYDERphd/polyBERT \
+  --revision "$revision" --local-dir "$polybert"
+
+worker() {
+  gpu=$1
+  shift
+  for seed in "$@"; do
+    result=$run_root/results/seed_${seed}.json
+    [[ -s $result ]] && continue
+    CUDA_VISIBLE_DEVICES=$gpu "$python_bin" scripts/train_polyedit_multiobjective.py \
+      --seed "$seed" --polybert_path "$polybert" --out "$result" \
+      --ckpt "$ckpt_root/seed_${seed}" --wandb 2>&1 | tee "$run_root/logs/seed_${seed}.log"
+  done
+}
+
+cd "$repo"
+worker 0 1004 1006 1008 &
+worker0=$!
+worker 1 1005 1007 &
+worker1=$!
+wait "$worker0" "$worker1"
+
+"$python_bin" scripts/summarize_polyedit_multiobjective.py \
+  --input "$run_root/results" --output "$run_root/summary.json" \
+  --markdown "$run_root/summary.md"