unparallelled commited on
Commit
08fbf4f
·
verified ·
1 Parent(s): a5106fe

Upload logs/mix64k/probe.log with huggingface_hub

Browse files
Files changed (1) hide show
  1. logs/mix64k/probe.log +61 -0
logs/mix64k/probe.log ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ === Llama Context Parallel Training Script (bluscriptCP) ===
2
+ [CP attn fusion] split (Stage-2 rope-once + Stage-3 Hopper causal-attn kernels)
3
+ [watchdog] collective no-progress deadline 600 s, poll 2 s (BLU_NCCL_WATCHDOG=0 to disable)
4
+ [watchdog] collective no-progress deadline 600 s, poll 2 s (BLU_NCCL_WATCHDOG=0 to disable)
5
+ [watchdog] collective no-progress deadline 600 s, poll 2 s (BLU_NCCL_WATCHDOG=0 to disable)
6
+ ================ Configuration (bluscriptCP) ================
7
+ d_model/layers : 1536 / 36
8
+ q/kv heads : 12 / 4 (G=3, hd=128)
9
+ ffn_hidden : 4096 (SwiGLU)
10
+ context_length : 65536
11
+ vocab_size : 50304 tying=0
12
+ B x T : 1 x 65536
13
+ world_size : 8 (dp=2 x cp=4)
14
+ CP_ATTN_MODE : ring (load_balance=HeadTail)
15
+ global_batch : 131072 (grad_accum=1)
16
+ precision : bf16 Linears + fp32 norms/residual/logits (fp32 master weights in AdamW) [mask=15 QKV WO MLP HEAD]
17
+ max_lr/min_lr : 0.0006 / 6e-05
18
+ warmup/steps : 100 / 3 (cosine, no stable phase)
19
+ checkpointing : false (ckpt_freq=100, dir=checkpoints_bluscriptcp)
20
+ =============================================================
21
+ Initializing model on CUDA device 0...
22
+ [YaRN] build_rope_cache: seq_len=65536 hd=128 scale=2.000 low=24 high=42 m=1.0693
23
+ [InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/rank_0/ext64k_step695.ckpt (base step 695, loss 2.93407) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
24
+ Parameters: 1060624896
25
+ Parameters per GPU: 1060624896
26
+ max_steps: 3
27
+ warmup_steps: 100
28
+ [ZeRO-1 v2] build=v2.4 supports: rs+ag | can-symmetric-window | can-fp32-reduce | can-bf16-reduce | can-reshard-v1-ckpt
29
+ [ZeRO-1 v2] ACTIVE: reduce_dtype=bfloat16 (= param dtype) symmetric=off overlap=-/-
30
+ [ZeRO-1 v2] zero1-v2 reduce-scatter/all-gather (standard memory)
31
+ [ZeRO-1 v2] params=983236608 padded=983236608 W=8 shard=122904576 | 38 buckets of 50 MB (3276800 elems/rank/bucket) | optimizer state/GPU: 1406.53 MB (vs 11252.2 MB unsharded)
32
+ [ZeRO-1 v2] NOTE: symmetric memory is NOT active — collectives will run NCCL's ring/tree path, not NVLS. Create the process group with MemoryMode::Symmetric (and NCCL >= 2.27) to enable it.
33
+ [ZeRO-1 v2] build=v2.4 supports: rs+ag | can-symmetric-window | can-fp32-reduce | can-bf16-reduce | can-reshard-v1-ckpt
34
+ [ZeRO-1 v2] ACTIVE: reduce_dtype=float32 (= param dtype) symmetric=off overlap=-/-
35
+ [ZeRO-1 v2] zero1-v2 reduce-scatter/all-gather (standard memory)
36
+ [ZeRO-1 v2] params=77388288 padded=77388288 W=8 shard=9673536 | 6 buckets of 50 MB (1638400 elems/rank/bucket) | optimizer state/GPU: 110.705 MB (vs 885.639 MB unsharded)
37
+ [ZeRO-1 v2] NOTE: symmetric memory is NOT active — collectives will run NCCL's ring/tree path, not NVLS. Create the process group with MemoryMode::Symmetric (and NCCL >= 2.27) to enable it.
38
+ [ZeRO-1] variant=v2 reduce-scatter/all-gather groups=2 [bfloat16/983236608] [float32/77388288]
39
+ [ZeRO-1] DataParallel is DISABLED (both would double-reduce)
40
+ [precision] bf16 params=217, all with fp32 master weights
41
+ found 10 shards for split train
42
+ found 1 shards for split val
43
+ Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log1.csv (run 1)
44
+ [rope_theta] 500000 compiled default (no sidecar -- legacy run, assuming it was trained at this base)
45
+ Checkpointing: OFF
46
+ [re-warmup] anchored at step 0 for 100 steps -> peak 0.00018 (0.3 x max_lr), then cosine-decay to 6e-05 | resuming at step 0 (mid-ramp)
47
+
48
+ Starting training...
49
+
50
+ [CP ring P2P] OVERLAP: dedicated non-blocking stream
51
+ validation loss: 3.0638
52
+ [CP bwd] overlap=ON
53
+ step 0 | loss: 2.895222 | lr 1.8000e-06 | norm: 0.3380 | dt: 18367.40ms | tok/sec: 7136.1 | mem: 97875MB | Time Left: 00 hrs : 00 mins
54
+ [TIMING] data: 0.4ms | fwd: 4960.0ms | loss: 0.9ms | bwd: 1382.4ms | clip: 0.0ms | optim: 22.0ms
55
+ step 1 | loss: 3.064343 | lr 3.6000e-06 | norm: 0.2482 | dt: 6899.80ms | tok/sec: 18996.5 | mem: 105619MB | Time Left: 00 hrs : 00 mins
56
+ [TIMING] data: 0.7ms | fwd: 5332.9ms | loss: 0.8ms | bwd: 1272.4ms | clip: 0.0ms | optim: 21.6ms
57
+ validation loss: 3.0637
58
+ step 2 | loss: 3.041020 | lr 5.4000e-06 | norm: 0.2996 | dt: 14879.81ms | tok/sec: 8808.7 | mem: 105747MB | Time Left: 00 hrs : 00 mins
59
+ [TIMING] data: 0.1ms | fwd: 4833.1ms | loss: 0.8ms | bwd: 1292.9ms | clip: 0.0ms | optim: 21.5ms
60
+
61
+ === Training complete ===