Upload folder using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_null_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_null_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_null_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b2.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b0.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b1.log +45 -0
- logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b2.log +45 -0
- logs/probe_run8/csv/ppl_needle_T4096_absent_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_absent_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_absent_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_null_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_null_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_null_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.25_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.25_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.25_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.5_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.5_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.5_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.75_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.75_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0.75_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p0_b2.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p1_b0.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p1_b1.csv +0 -0
- logs/probe_run8/csv/ppl_needle_T4096_p1_b2.csv +0 -0
- logs/probe_run8/needles/lam_ntk_hd128_th500000_s1.txt +4 -0
- logs/probe_run8/needles/needle_T4096_absent_b0.bin +3 -0
- logs/probe_run8/needles/needle_T4096_absent_b0.json +37 -0
- logs/probe_run8/needles/needle_T4096_absent_b1.bin +3 -0
- logs/probe_run8/needles/needle_T4096_absent_b1.json +37 -0
- logs/probe_run8/needles/needle_T4096_absent_b2.bin +3 -0
- logs/probe_run8/needles/needle_T4096_absent_b2.json +37 -0
- logs/probe_run8/needles/needle_T4096_null_b0.bin +3 -0
logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_absent_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_absent_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_absent_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_absent_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_absent_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_absent_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_absent_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_absent_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_absent_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_absent_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_null_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_null_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_null_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_null_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_null_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_null_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_null_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_null_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_null_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_null_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_null_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_null_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.25_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.25_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.25_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.25_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.25_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.25_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.25_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.25_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.25_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.25_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.5_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.5_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.5_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.5_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.5_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.5_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.5_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.5_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.5_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.5_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.75_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.75_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.75_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.75_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.75_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.75_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0.75_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0.75_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0.75_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0.75_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p0_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p0_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p0_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p0_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b0.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p1_b0.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p1_b0.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p1_b0.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b1.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p1_b1.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p1_b1.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p1_b1.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/1b_run8_T4096_ntk_p1_b2.log
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
=== Llama Context Parallel Training Script (bluscriptCP) ===
|
| 2 |
+
[CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
|
| 3 |
+
[LongRoPE] factors=probe_run8/needles/lam_ntk_hd128_th500000_s1.txt n_hat=0 s=1 S_search=4096 (alone)
|
| 4 |
+
================ Configuration (bluscriptCP) ================
|
| 5 |
+
d_model/layers : 1536 / 36
|
| 6 |
+
q/kv heads : 12 / 4 (G=3, hd=128)
|
| 7 |
+
ffn_hidden : 4096 (SwiGLU)
|
| 8 |
+
context_length : 4096
|
| 9 |
+
vocab_size : 50304 tying=0
|
| 10 |
+
B x T : 1 x 4096
|
| 11 |
+
world_size : 1 (dp=1 x cp=1)
|
| 12 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 13 |
+
global_batch : 524288 (grad_accum=128)
|
| 14 |
+
optimizer : AdamW [DDP]
|
| 15 |
+
precision : fp32 everywhere
|
| 16 |
+
max_lr/min_lr : 0.0006 / 6e-05
|
| 17 |
+
warmup/steps : 91 / 917 (cosine, no stable phase)
|
| 18 |
+
checkpointing : true (ckpt_freq=50, dir=checkpoints_bluscriptcp)
|
| 19 |
+
=============================================================
|
| 20 |
+
Initializing model on CUDA device 0...
|
| 21 |
+
[YaRN] build_rope_cache: seq_len=4096 hd=128 scale=1.000 low=7 high=25 m=1.0000
|
| 22 |
+
[LongRoPE] build_rope_cache_longrope: seq_len=4096 hd=128 n_hat=0 lam0=1.0000 lam_last=1.0000 mscale=1.0000
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/blumodelcp_run8_step_1000.ckpt (base step 1000, loss 2.66226) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 1060624896
|
| 25 |
+
Parameters per GPU: 1060624896
|
| 26 |
+
max_steps: 917
|
| 27 |
+
warmup_steps: 91
|
| 28 |
+
found 100 shards for split train
|
| 29 |
+
found 1 shards for split val
|
| 30 |
+
[Resume] latest run 8 already complete at step 1003; starting a new run.
|
| 31 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log7.csv (run 7)
|
| 32 |
+
[rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
|
| 33 |
+
Checkpointing: ON -> checkpoints_bluscriptcp/blumodelcp_run7_step_<N>.ckpt (every 50 steps, keep 6)
|
| 34 |
+
[eval] shared token stream: 32769 int32 tokens from probe_run8/needles/needle_T4096_p1_b2.bin (B forced to 1 for HF-scorer parity)
|
| 35 |
+
[eval] PPL-vs-position: 8 windows (skip=0) x B=1 Tglobal=4096 Tlocal=4096 cp=1 (YARN_SCALE via env) -> probe_run8/csv/ppl_needle_T4096_p1_b2.csv
|
| 36 |
+
[eval] per-position NLL: GPU
|
| 37 |
+
[eval] window 1/8 done
|
| 38 |
+
[eval] window 2/8 done
|
| 39 |
+
[eval] window 3/8 done
|
| 40 |
+
[eval] window 4/8 done
|
| 41 |
+
[eval] window 5/8 done
|
| 42 |
+
[eval] window 6/8 done
|
| 43 |
+
[eval] window 7/8 done
|
| 44 |
+
[eval] window 8/8 done
|
| 45 |
+
[eval] wrote probe_run8/csv/ppl_needle_T4096_p1_b2.csv (4096 positions, 8 windows)
|
logs/probe_run8/csv/ppl_needle_T4096_absent_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_absent_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_absent_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_null_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_null_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_null_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.25_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.25_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.25_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.5_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.5_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.5_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.75_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.75_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0.75_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p0_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p1_b0.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p1_b1.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/csv/ppl_needle_T4096_p1_b2.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
logs/probe_run8/needles/lam_ntk_hd128_th500000_s1.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
n_hat 0
|
| 2 |
+
s 1
|
| 3 |
+
S_search 4096
|
| 4 |
+
lambda 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00 1.00
|
logs/probe_run8/needles/needle_T4096_absent_b0.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9cb63d258f1d99f6da7541d9be6b7e03ad7fa0924eb231f5317c7a473d1645b7
|
| 3 |
+
size 131076
|
logs/probe_run8/needles/needle_T4096_absent_b0.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bin": "probe_run8/needles/needle_T4096_absent_b0.bin",
|
| 3 |
+
"variant": "absent",
|
| 4 |
+
"block": 0,
|
| 5 |
+
"p_frac": null,
|
| 6 |
+
"is_null": false,
|
| 7 |
+
"trials": 8,
|
| 8 |
+
"seed": 0,
|
| 9 |
+
"vocab": 50304,
|
| 10 |
+
"reverse": false,
|
| 11 |
+
"slot_starts": [
|
| 12 |
+
null,
|
| 13 |
+
null,
|
| 14 |
+
null,
|
| 15 |
+
null,
|
| 16 |
+
null,
|
| 17 |
+
null,
|
| 18 |
+
null,
|
| 19 |
+
null
|
| 20 |
+
],
|
| 21 |
+
"T": 4096,
|
| 22 |
+
"needle_len": 16,
|
| 23 |
+
"pad_tail": 4,
|
| 24 |
+
"probe_pos": null,
|
| 25 |
+
"probe_start": 4076,
|
| 26 |
+
"answer_start": 4084,
|
| 27 |
+
"answer_end": 4092,
|
| 28 |
+
"max_slot_start": 4060,
|
| 29 |
+
"answer_offsets": [
|
| 30 |
+
4084,
|
| 31 |
+
4092
|
| 32 |
+
],
|
| 33 |
+
"csv_positions": [
|
| 34 |
+
4083,
|
| 35 |
+
4091
|
| 36 |
+
]
|
| 37 |
+
}
|
logs/probe_run8/needles/needle_T4096_absent_b1.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bf9849449af5338369d5a58c1e8df9b431ffbeea7b16ee261af2896438c7f050
|
| 3 |
+
size 131076
|
logs/probe_run8/needles/needle_T4096_absent_b1.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bin": "probe_run8/needles/needle_T4096_absent_b1.bin",
|
| 3 |
+
"variant": "absent",
|
| 4 |
+
"block": 1,
|
| 5 |
+
"p_frac": null,
|
| 6 |
+
"is_null": false,
|
| 7 |
+
"trials": 8,
|
| 8 |
+
"seed": 0,
|
| 9 |
+
"vocab": 50304,
|
| 10 |
+
"reverse": false,
|
| 11 |
+
"slot_starts": [
|
| 12 |
+
null,
|
| 13 |
+
null,
|
| 14 |
+
null,
|
| 15 |
+
null,
|
| 16 |
+
null,
|
| 17 |
+
null,
|
| 18 |
+
null,
|
| 19 |
+
null
|
| 20 |
+
],
|
| 21 |
+
"T": 4096,
|
| 22 |
+
"needle_len": 16,
|
| 23 |
+
"pad_tail": 4,
|
| 24 |
+
"probe_pos": null,
|
| 25 |
+
"probe_start": 4076,
|
| 26 |
+
"answer_start": 4084,
|
| 27 |
+
"answer_end": 4092,
|
| 28 |
+
"max_slot_start": 4060,
|
| 29 |
+
"answer_offsets": [
|
| 30 |
+
4084,
|
| 31 |
+
4092
|
| 32 |
+
],
|
| 33 |
+
"csv_positions": [
|
| 34 |
+
4083,
|
| 35 |
+
4091
|
| 36 |
+
]
|
| 37 |
+
}
|
logs/probe_run8/needles/needle_T4096_absent_b2.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99978ca47137c8ade4bb44f3e4979e7dabaaf2a7e2118b399f3eb9291983d455
|
| 3 |
+
size 131076
|
logs/probe_run8/needles/needle_T4096_absent_b2.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bin": "probe_run8/needles/needle_T4096_absent_b2.bin",
|
| 3 |
+
"variant": "absent",
|
| 4 |
+
"block": 2,
|
| 5 |
+
"p_frac": null,
|
| 6 |
+
"is_null": false,
|
| 7 |
+
"trials": 8,
|
| 8 |
+
"seed": 0,
|
| 9 |
+
"vocab": 50304,
|
| 10 |
+
"reverse": false,
|
| 11 |
+
"slot_starts": [
|
| 12 |
+
null,
|
| 13 |
+
null,
|
| 14 |
+
null,
|
| 15 |
+
null,
|
| 16 |
+
null,
|
| 17 |
+
null,
|
| 18 |
+
null,
|
| 19 |
+
null
|
| 20 |
+
],
|
| 21 |
+
"T": 4096,
|
| 22 |
+
"needle_len": 16,
|
| 23 |
+
"pad_tail": 4,
|
| 24 |
+
"probe_pos": null,
|
| 25 |
+
"probe_start": 4076,
|
| 26 |
+
"answer_start": 4084,
|
| 27 |
+
"answer_end": 4092,
|
| 28 |
+
"max_slot_start": 4060,
|
| 29 |
+
"answer_offsets": [
|
| 30 |
+
4084,
|
| 31 |
+
4092
|
| 32 |
+
],
|
| 33 |
+
"csv_positions": [
|
| 34 |
+
4083,
|
| 35 |
+
4091
|
| 36 |
+
]
|
| 37 |
+
}
|
logs/probe_run8/needles/needle_T4096_null_b0.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f3625f3f89f1599ee43b8b2ba0f5d584a6f969d5a3fdcbd679ca47dab111df53
|
| 3 |
+
size 131076
|