Upload scripts/eval8.sh with huggingface_hub
Browse files- scripts/eval8.sh +27 -0
scripts/eval8.sh
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/usr/bin/env bash
|
| 2 |
+
# 2 checkpoints x 4 lengths = 8 jobs, one per GPU, all 8 cards busy.
|
| 3 |
+
# Same wall time as 4 jobs (each length still runs its ckpts in parallel) but
|
| 4 |
+
# yields a 300-vs-500 trajectory instead of a single point.
|
| 5 |
+
set -uo pipefail; cd "$(dirname "$0")"
|
| 6 |
+
export PATH=/root/Adhi/BluTrain/envv/bin:$PATH
|
| 7 |
+
source /root/Adhi/BluTrain/envv/bin/activate 2>/dev/null
|
| 8 |
+
L=t128b_logs; M="$L/eval8.log"
|
| 9 |
+
say(){ echo "[$(date -u +%H:%M:%S)] $*" | tee -a "$M" >&2; }
|
| 10 |
+
A680="CP_N_EMBD=1536 CP_N_LAYER=24 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=1"
|
| 11 |
+
HAY=Data_Loader/Data/b2_8192_16384/val_shard_00000.bin
|
| 12 |
+
blk(){ local PL=$1 T=$2 G=$3 O="e8_$1"
|
| 13 |
+
rm -rf "$O"
|
| 14 |
+
A1B="$A680" HAY="$HAY" GEN_DATA_ROOT="Data_Loader/Data/evalsrc" \
|
| 15 |
+
CKPT_EXTRA="s300=t128b_s300.ckpt s500=t128b_s500.ckpt" CKPTS="s300 s500" \
|
| 16 |
+
TASKS="niah_single" N=100 T=$T YARN=32 YARN_ORIG=4096 PROMPT_LEN=$PL \
|
| 17 |
+
GPUS="$G" NP=1 OUT="$O" ./run_longeval.sh > "$L/e8_${PL}.log" 2>&1
|
| 18 |
+
say " len $PL $(head -1 $O/summary.csv 2>/dev/null) | $(grep -h '^niah_single,' $O/summary.csv 2>/dev/null)"; }
|
| 19 |
+
say "=== eval: s300 + s500, 4 lengths, 8 GPUs ==="
|
| 20 |
+
blk 65512 65536 "0,1" & blk 32744 32768 "2,3" & blk 16360 16384 "4,5" & blk 4096 8192 "6,7" &
|
| 21 |
+
wait
|
| 22 |
+
say "=== RESULTS (cols per file: see header) vs t64@x32 and old-t128@x32 ==="
|
| 23 |
+
for c in 4096:100:98 16360:99:98 32744:93:69 65512:63:31; do
|
| 24 |
+
PL=${c%%:*}; r=${c#*:}; t64=${r%%:*}; old=${r##*:}
|
| 25 |
+
say " $PL: $(head -1 e8_${PL}/summary.csv 2>/dev/null) -> $(grep -h '^niah_single,' e8_${PL}/summary.csv 2>/dev/null) [t64=$t64 old128=$old]"
|
| 26 |
+
done
|
| 27 |
+
say "EVAL8 DONE"
|