unparallelled commited on
Commit
c9544dc
·
verified ·
1 Parent(s): a48ce41

Upload scripts/t64_finish.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 ***"