unparallelled commited on
Commit
068ae50
·
verified ·
1 Parent(s): c47537e

Upload scripts/eval_scale1.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. scripts/eval_scale1.sh +32 -0
scripts/eval_scale1.sh ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ # THE DECISIVE TEST: YaRN OFF (scale=1). virtual position == real position.
3
+ # Everything inside the 4,096 pretrain range, no compression, no interpolation.
4
+ #
5
+ # Every score we have is a function of VIRTUAL position and not token count:
6
+ # v128 -> ~99, v1023 -> ~92, v2047 -> 45-63, v4094 -> 41
7
+ # and t128c EQUALS t64 at v128 and v1023. So extension training added no new
8
+ # positional competence; only inference-time compression ever helped.
9
+ #
10
+ # If these models fail at REAL position 2,048-4,096 with scale=1, the ceiling is
11
+ # in the base model's pretraining (theta=500,000 at a 4,096 pretrain leaves 32 of
12
+ # 64 RoPE dims without a full rotation) and no extension recipe can fix it.
13
+ # If they score ~95 flat, positions 1-4096 are healthy and YaRN's ramp at
14
+ # scale>1 is what destroys v2047 -- a fixable extension problem.
15
+ set -uo pipefail; cd "$(dirname "$0")"
16
+ export LD_LIBRARY_PATH="$(cd ../.. && pwd)/Tensor-Implementations/lib:$(cd ../.. && pwd)/Profiler/lib:${LD_LIBRARY_PATH:-}"
17
+ export PATH=/usr/local/cuda/bin:$PATH; export PY=/root/Adhi/BluTrain/envv/bin/python
18
+ L=eval_scale1_logs; mkdir -p "$L"
19
+ A680="CP_N_EMBD=1536 CP_N_LAYER=24 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=1"
20
+ say(){ echo "[$(date -u +%H:%M:%S)] $*" | tee -a "$L/master.log" >&2; }
21
+ : # ordering handled by eval_chain.sh
22
+ say "=== scale=1 (YaRN OFF): p1_s200 / t64 / c300 at 1024, 2048, 4096 ==="
23
+ for PL in 1024 2048 4096; do
24
+ O="es1_${PL}"
25
+ A1B="$A680" HAY="data/mix128_b6_60/val_shard_00000.bin" GEN_DATA_ROOT="data/mix128_b6_60" \
26
+ CKPT_EXTRA="p1=p1_s200.ckpt t64=t64.ckpt c300=t128c_s300.ckpt" CKPTS="p1 t64 c300" \
27
+ TASKS="niah_single" N=100 T=8192 YARN=1 YARN_ORIG=4096 PROMPT_LEN="$PL" \
28
+ GPUS="0,1,2,3" NP=1 JOBS_PER_GPU=3 OUT="$O" ./run_longeval.sh > "$L/len_${PL}.log" 2>&1
29
+ say " real pos $PL: $(head -1 $O/summary.csv 2>/dev/null|tr -d '\r') -> $(grep -h '^niah_single,' $O/summary.csv 2>/dev/null|tr -d '\r')"
30
+ done
31
+ say "compare: at virtual 1023 every model scores ~92; at v2047, 45-63"
32
+ say "EVAL_SCALE1_DONE"