Upload scripts/t64_finish.sh with huggingface_hub
Browse files- scripts/t64_finish.sh +73 -0
scripts/t64_finish.sh
ADDED
|
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
# Wait for the 64k run -> three-way eval -> upload verified -> teardown verdict.
|
| 3 |
+
set -uo pipefail
|
| 4 |
+
cd "$(dirname "$0")"
|
| 5 |
+
export PATH=/root/Adhi/BluTrain/envv/bin:$PATH
|
| 6 |
+
source /root/Adhi/BluTrain/envv/bin/activate 2>/dev/null
|
| 7 |
+
REPO=unparallelled/BluScriptCP
|
| 8 |
+
L=t64_logs; M="$L/finish.log"
|
| 9 |
+
say(){ echo "[$(date -u +%H:%M:%S)] $*" | tee -a "$M" >&2; }
|
| 10 |
+
|
| 11 |
+
say "waiting for training to finish writing..."
|
| 12 |
+
while ! grep -qa '64K TRAINING COMPLETE\|FATAL' "$L/master.log" 2>/dev/null; do
|
| 13 |
+
pgrep -f bluscriptCP_exec >/dev/null 2>&1 || { sleep 40
|
| 14 |
+
grep -qa '64K TRAINING COMPLETE\|FATAL' "$L/master.log" 2>/dev/null || break; }
|
| 15 |
+
sleep 30
|
| 16 |
+
done
|
| 17 |
+
grep -qa FATAL "$L/master.log" && say "training FAILED"
|
| 18 |
+
|
| 19 |
+
CK=checkpoints_bluscriptcp/rank_0
|
| 20 |
+
RN=$(cat "$L/runnum" 2>/dev/null || echo "")
|
| 21 |
+
FINAL=""
|
| 22 |
+
[ -n "$RN" ] && FINAL=$(ls -t $CK/blumodelcp_run${RN}_step_*.ckpt 2>/dev/null | head -1)
|
| 23 |
+
[ -z "$FINAL" ] && FINAL=$(ls -t $CK/blumodelcp_run*_step_*.ckpt 2>/dev/null | head -1)
|
| 24 |
+
[ -z "$FINAL" ] && { say "NO CHECKPOINT -- cannot evaluate"; exit 1; }
|
| 25 |
+
say "final: $(basename "$FINAL") $(stat -c%s "$FINAL") B"
|
| 26 |
+
|
| 27 |
+
A680="CP_N_EMBD=1536 CP_N_LAYER=24 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=1"
|
| 28 |
+
HAY=Data_Loader/Data/b2_8192_16384/val_shard_00000.bin
|
| 29 |
+
[ "$(stat -c%s "$HAY")" = 199754694 ] || { say "FATAL haystack differs from every prior eval"; exit 1; }
|
| 30 |
+
EX="t64=$(basename "$FINAL") p1final=p1final.ckpt base680=varybatch.ckpt"
|
| 31 |
+
ev(){ # T SCALE PL TASKS tag
|
| 32 |
+
local T=$1 S=$2 PL=$3 TK=$4 TG=$5
|
| 33 |
+
local O="t64ev_${TG}_${PL}"
|
| 34 |
+
[ -s "$O/summary.csv" ] && [ "$(wc -l < "$O/summary.csv")" -gt 1 ] && { say " $TG $PL cached"; return 0; }
|
| 35 |
+
say " eval $TG len $PL (T=$T x$S -> virtual $((PL/S))) [$TK]"
|
| 36 |
+
A1B="$A680" HAY="$HAY" GEN_DATA_ROOT="Data_Loader/Data/evalsrc" \
|
| 37 |
+
CKPT_EXTRA="$EX" CKPTS="t64 p1final base680" TASKS="$TK" N=100 \
|
| 38 |
+
T=$T YARN=$S YARN_ORIG=4096 PROMPT_LEN=$PL GPUS=0,1,2,3,4,5,6,7 NP=1 OUT="$O" \
|
| 39 |
+
./run_longeval.sh >> "$L/ev_${TG}_${PL}.log" 2>&1
|
| 40 |
+
if [ -s "$O/summary.csv" ] && [ "$(wc -l < "$O/summary.csv")" -gt 1 ]; then
|
| 41 |
+
say " $(head -1 $O/summary.csv)"; tail -n +2 "$O/summary.csv"|sed 's/^/ /'|tee -a "$M" >&2
|
| 42 |
+
else say " FAILED"; grep -ahA2 'what()' "$O"/gen/*.log 2>/dev/null|head -3|tee -a "$M" >&2; fi
|
| 43 |
+
}
|
| 44 |
+
say "=== THREE-WAY EVAL: t64 vs p1final vs base680 ==="
|
| 45 |
+
ev 262144 64 4096 niah_single x64
|
| 46 |
+
ev 262144 64 32744 niah_single x64
|
| 47 |
+
ev 262144 64 65512 niah_single x64
|
| 48 |
+
ev 65536 16 65512 niah_single x16at
|
| 49 |
+
say "=== harder tasks -- if NIAH rises and these do not, it is a benchmark gain ==="
|
| 50 |
+
ev 262144 64 32744 "niah_multikey val_backward" hard
|
| 51 |
+
|
| 52 |
+
up(){ [ -f "$1" ] || { say " MISSING $1"; return 1; }
|
| 53 |
+
hf upload "$REPO" "$1" "$2" >>"$L/upload.log" 2>&1
|
| 54 |
+
python - "$REPO" "$1" "$2" <<'V'
|
| 55 |
+
import sys,os
|
| 56 |
+
from huggingface_hub import HfApi
|
| 57 |
+
w=os.path.getsize(sys.argv[2])
|
| 58 |
+
g={f.rfilename:(f.size or 0) for f in HfApi().repo_info(sys.argv[1],files_metadata=True).siblings}.get(sys.argv[3])
|
| 59 |
+
ok=(g==w); print((" VERIFIED " if ok else " *** FAILED ")+f"{sys.argv[3]} local={w:,} remote={g}")
|
| 60 |
+
sys.exit(0 if ok else 1)
|
| 61 |
+
V
|
| 62 |
+
}
|
| 63 |
+
BAD=0
|
| 64 |
+
say "=== UPLOAD ==="
|
| 65 |
+
for f in t64_train.sh t64_finish.sh dl_64k.sh; do up "$f" "scripts/$f" || BAD=$((BAD+1)); done
|
| 66 |
+
up Data_Loader/Data/mix64/MANIFEST.txt data/mix64_MANIFEST.txt || BAD=$((BAD+1))
|
| 67 |
+
for g in master.log train.log; do cp -f "$L/$g" "$L/$g.snap"; up "$L/$g.snap" "logs/t64_$g" || BAD=$((BAD+1)); done
|
| 68 |
+
for d in t64ev_*; do [ -f "$d/summary.csv" ] || continue
|
| 69 |
+
up "$d/summary.csv" "evals/2026-09-17/${d}_summary.csv" || BAD=$((BAD+1)); done
|
| 70 |
+
ST=$(basename "$FINAL" | grep -oE 'step_[0-9]+' | cut -d_ -f2)
|
| 71 |
+
up "$FINAL" "checkpoints/t64_680m_step${ST}_rank0.ckpt" || BAD=$((BAD+1))
|
| 72 |
+
say "=== TEARDOWN VERDICT ==="
|
| 73 |
+
[ "$BAD" -eq 0 ] && say "ALL VERIFIED -- SAFE TO TEAR DOWN" || say "*** $BAD failure(s) -- DO NOT TEAR DOWN ***"
|