Upload scripts/final_evals.sh with huggingface_hub
Browse files- scripts/final_evals.sh +51 -0
scripts/final_evals.sh
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
# 1. x64 sweep -- does the scale curve keep rising at 64k?
|
| 3 |
+
# 2. val_backward -- the DISCRIMINATING task. 2026-09-09 found niah_single is
|
| 4 |
+
# non-discriminating (Qwen3 0.6B-8B all 100%, zero variation over 13x params)
|
| 5 |
+
# while val_backward separates them and ours were at chance. If today's NIAH
|
| 6 |
+
# gains are real capability, val_backward should move too. If they are a
|
| 7 |
+
# compression artefact, it will not.
|
| 8 |
+
set -uo pipefail
|
| 9 |
+
cd "$(dirname "$0")"
|
| 10 |
+
export PATH=/root/Adhi/BluTrain/envv/bin:$PATH
|
| 11 |
+
source /root/Adhi/BluTrain/envv/bin/activate 2>/dev/null
|
| 12 |
+
L=final_logs; mkdir -p "$L"; M="$L/master.log"
|
| 13 |
+
say(){ echo "[$(date -u +%H:%M:%S)] $*" | tee -a "$M" >&2; }
|
| 14 |
+
HAY=Data_Loader/Data/b2_8192_16384/val_shard_00000.bin
|
| 15 |
+
[ "$(stat -c%s "$HAY")" = 199754694 ] || { say "FATAL haystack differs"; exit 1; }
|
| 16 |
+
A680="CP_N_EMBD=1536 CP_N_LAYER=24 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=1"
|
| 17 |
+
|
| 18 |
+
go(){ # T SCALE PL TASKS CKPTS EXTRA ARCH tag
|
| 19 |
+
local T=$1 S=$2 PL=$3 TK=$4 CKS=$5 EX=$6 AR=$7 TAG=$8
|
| 20 |
+
local O="fin_${TAG}_${PL}"
|
| 21 |
+
[ $((S*4096)) -eq $T ] || { say "YaRN invariant $S*4096 != $T"; return 1; }
|
| 22 |
+
[ -s "$O/summary.csv" ] && [ "$(wc -l < "$O/summary.csv")" -gt 1 ] && {
|
| 23 |
+
say " $TAG $PL cached: $(tail -n +2 $O/summary.csv|tr '\n' ' ')"; return 0; }
|
| 24 |
+
say " $TAG len $PL (T=$T x$S -> virtual $((PL/S))) tasks=$TK"
|
| 25 |
+
A1B="$AR" HAY="$HAY" GEN_DATA_ROOT="Data_Loader/Data/evalsrc" \
|
| 26 |
+
CKPT_EXTRA="$EX" CKPTS="$CKS" TASKS="$TK" N=100 \
|
| 27 |
+
T=$T YARN=$S YARN_ORIG=4096 PROMPT_LEN=$PL GPUS=0,1 NP=1 OUT="$O" \
|
| 28 |
+
./run_longeval.sh >> "$L/${TAG}_${PL}.log" 2>&1
|
| 29 |
+
if [ -s "$O/summary.csv" ] && [ "$(wc -l < "$O/summary.csv")" -gt 1 ]; then
|
| 30 |
+
say " $(head -1 $O/summary.csv)"; tail -n +2 "$O/summary.csv" | sed 's/^/ /' | tee -a "$M" >&2
|
| 31 |
+
else say " FAILED"; grep -ahA2 'what()' "$O"/gen/*.log 2>/dev/null|head -3|tee -a "$M" >&2; fi
|
| 32 |
+
}
|
| 33 |
+
E680="p1final=blumodelcp_run12_step_1200.ckpt base680=varybatch.ckpt"
|
| 34 |
+
|
| 35 |
+
say "=== PART 1: x64 sweep (T=262144) -- does scale still help at 64k? ==="
|
| 36 |
+
for PL in 4096 32744 65512; do go 262144 64 $PL niah_single "p1final base680" "$E680" "$A680" x64; done
|
| 37 |
+
|
| 38 |
+
say "=== PART 2: val_backward at the best configs -- the discriminating task ==="
|
| 39 |
+
go 65536 16 16360 "val_backward niah_multikey" "p1final base680" "$E680" "$A680" vb16
|
| 40 |
+
go 131072 32 32744 "val_backward niah_multikey" "p1final base680" "$E680" "$A680" vb32
|
| 41 |
+
say " and the 1.06B (trained to 64k) on the same task -- default 36-layer arch"
|
| 42 |
+
E106="mix1200=mix64k_step_1200.ckpt"
|
| 43 |
+
A106="CP_N_EMBD=1536 CP_N_LAYER=36 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=0"
|
| 44 |
+
go 131072 32 32744 "val_backward niah_multikey" "mix1200" "$E106" "$A106" vb106
|
| 45 |
+
|
| 46 |
+
say "=== ALL RESULTS ==="
|
| 47 |
+
for d in fin_*; do
|
| 48 |
+
[ -f "$d/summary.csv" ] || continue
|
| 49 |
+
say " $d: $(head -1 $d/summary.csv) | $(tail -n +2 $d/summary.csv | tr '\n' ' ')"
|
| 50 |
+
done
|
| 51 |
+
say "DONE"
|