diff options
| author | Yuren Hao <yurenh2@illinois.edu> | 2026-07-04 23:04:53 -0500 |
|---|---|---|
| committer | Yuren Hao <yurenh2@illinois.edu> | 2026-07-04 23:04:53 -0500 |
| commit | 1cea113ef78d2703b024a088703a06ac5d235c5c (patch) | |
| tree | 53e9f829f0ba59d2c7ecc08c132349498ef361a7 /ep_run/runs | |
| parent | 873fff62ad23f8ee81e0944a6de67f7215619d71 (diff) | |
queue v3: 5th arm abl_delay_hr2 (redx-faithful hr0.2 front) + every-100 saves on delay arms for dip-screening
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_014FAPDWQ49M5Ye3NpTndTpn
Diffstat (limited to 'ep_run/runs')
| -rwxr-xr-x | ep_run/runs/abl3_queue.sh | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/ep_run/runs/abl3_queue.sh b/ep_run/runs/abl3_queue.sh index 6fc47ad..bf2fda4 100755 --- a/ep_run/runs/abl3_queue.sh +++ b/ep_run/runs/abl3_queue.sh @@ -8,6 +8,9 @@ # arm1 abl_pair — proven pair from step 0 (resreg 0.2 + FROZEN jr 0.1; ★2.09 recipe) = delay's control # arm2 abl_floss — floss-only (graded finite-horizon LE penalty), --noadaptc (hidden damping feedback OFF) # arm3 abl_resreg — resreg-only, --noadaptc +# arm4 abl_delay_hr2 — FULL redx replication front: hr 0.2 (redx & warm_fast's holo rate; canonical is +# 0.02) + reg-free 2000 then pair. Pins the hr second-variable if arm0 fails to +# reproduce the edge-riding drift. # All arms share every other flag with the proven ep_resreg_scratch cmd (EP_BELOW210:97-101), same seed. # Queue: poll GPUs 0/1/3 (GPU2 = japardi2 NV-Embed server, DO NOT TOUCH); a slot is free when # mem.used < 38 GB AND util < 30% for 3 consecutive 60 s polls; launch the next arm per freed slot. @@ -17,7 +20,7 @@ echo "[$(date)] queue runner v2 up (pid $$, 4 arms, delay-first)" >> "$LOG" launch () { # $1 = gpu id, $2 = arm index case $2 in - 0) CUDA_VISIBLE_DEVICES=$1 nohup python3 lt_ep_train.py --mode ep --attn_mode thick --B 24 --C 512 --H 16 --T 256 --c 1.0 --jacreg 0.1 --jr_floor 0.1 --jr_max 0.1 --resreg 0.2 --reg_delay 2000 --holo 2 --hr 0.02 --t2sel 40 --track --pema 0.999 --t1max 300 --res_est 1e-4 --res_gate 0 --qknorm --resinit 0.1 --warmup 800 --T1 150 --T2 20 --lr 6e-4 --wsd 0.25 --steps 32000 --log 200 --save_every 500 --abort_res 0.3 --data data/tinystories_bpe --ckpt runs/abl_delay.pt --state runs/abl_delay.state > runs/abl_delay.log 2>&1 & + 0) CUDA_VISIBLE_DEVICES=$1 nohup python3 lt_ep_train.py --mode ep --attn_mode thick --B 24 --C 512 --H 16 --T 256 --c 1.0 --jacreg 0.1 --jr_floor 0.1 --jr_max 0.1 --resreg 0.2 --reg_delay 2000 --holo 2 --hr 0.02 --t2sel 40 --track --pema 0.999 --t1max 300 --res_est 1e-4 --res_gate 0 --qknorm --resinit 0.1 --warmup 800 --T1 150 --T2 20 --lr 6e-4 --wsd 0.25 --steps 32000 --log 100 --save_every 100 --abort_res 0.3 --data data/tinystories_bpe --ckpt runs/abl_delay.pt --state runs/abl_delay.state > runs/abl_delay.log 2>&1 & echo "[$(date)] abl_delay (reg-free 2k -> pair) -> GPU$1 pid $!" >> "$LOG" ;; 1) CUDA_VISIBLE_DEVICES=$1 nohup python3 lt_ep_train.py --mode ep --attn_mode thick --B 24 --C 512 --H 16 --T 256 --c 1.0 --jacreg 0.1 --jr_floor 0.1 --jr_max 0.1 --resreg 0.2 --holo 2 --hr 0.02 --t2sel 40 --track --pema 0.999 --t1max 300 --res_est 1e-4 --res_gate 0 --qknorm --resinit 0.1 --warmup 800 --T1 150 --T2 20 --lr 6e-4 --wsd 0.25 --steps 32000 --log 200 --save_every 500 --abort_res 0.3 --data data/tinystories_bpe --ckpt runs/abl_pair.pt --state runs/abl_pair.state > runs/abl_pair.log 2>&1 & echo "[$(date)] abl_pair (proven pair, control) -> GPU$1 pid $!" >> "$LOG" ;; @@ -25,14 +28,16 @@ launch () { # $1 = gpu id, $2 = arm index echo "[$(date)] abl_floss (floss-only, noadaptc) -> GPU$1 pid $!" >> "$LOG" ;; 3) CUDA_VISIBLE_DEVICES=$1 nohup python3 lt_ep_train.py --mode ep --attn_mode thick --B 24 --C 512 --H 16 --T 256 --c 1.0 --jacreg 0 --resreg 0.2 --noadaptc --holo 2 --hr 0.02 --t2sel 40 --track --pema 0.999 --t1max 300 --res_est 1e-4 --res_gate 0 --qknorm --resinit 0.1 --warmup 800 --T1 150 --T2 20 --lr 6e-4 --wsd 0.25 --steps 32000 --log 200 --save_every 500 --abort_res 0.3 --data data/tinystories_bpe --ckpt runs/abl_resreg.pt --state runs/abl_resreg.state > runs/abl_resreg.log 2>&1 & echo "[$(date)] abl_resreg (resreg-only, noadaptc) -> GPU$1 pid $!" >> "$LOG" ;; + 4) CUDA_VISIBLE_DEVICES=$1 nohup python3 lt_ep_train.py --mode ep --attn_mode thick --B 24 --C 512 --H 16 --T 256 --c 1.0 --jacreg 0.1 --jr_floor 0.1 --jr_max 0.1 --resreg 0.2 --reg_delay 2000 --holo 2 --hr 0.2 --t2sel 40 --track --pema 0.999 --t1max 300 --res_est 1e-4 --res_gate 0 --qknorm --resinit 0.1 --warmup 800 --T1 150 --T2 20 --lr 6e-4 --wsd 0.25 --steps 32000 --log 100 --save_every 100 --abort_res 0.3 --data data/tinystories_bpe --ckpt runs/abl_delay_hr2.pt --state runs/abl_delay_hr2.state > runs/abl_delay_hr2.log 2>&1 & + echo "[$(date)] abl_delay_hr2 (redx-faithful front: hr0.2 + reg-free 2k -> pair) -> GPU$1 pid $!" >> "$LOG" ;; esac } i=0 declare -A CNT USED -while [ $i -lt 4 ]; do +while [ $i -lt 5 ]; do for g in 0 1 3; do - [ $i -ge 4 ] && break + [ $i -ge 5 ] && break [ -n "${USED[$g]}" ] && continue read -r mem util <<< "$(nvidia-smi --query-gpu=memory.used,utilization.gpu --format=csv,noheader,nounits -i "$g" 2>/dev/null | awk -F',' '{gsub(/ /,""); print $1" "$2}')" if [ -n "$mem" ] && [ "$mem" -lt 38000 ] && [ "$util" -lt 30 ] 2>/dev/null; then @@ -49,4 +54,4 @@ while [ $i -lt 4 ]; do done sleep 60 done -echo "[$(date)] all four arms launched — queue runner exiting" >> "$LOG" +echo "[$(date)] all five arms launched — queue runner exiting" >> "$LOG" |
