unparallelled commited on
Commit
47f4c1d
·
verified ·
1 Parent(s): a5e1544

Upload logs/p1_probe.log with huggingface_hub

Browse files
Files changed (1) hide show
  1. logs/p1_probe.log +60 -0
logs/p1_probe.log ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ================ Configuration (bluscriptCP) ================
6
+ d_model/layers : 1536 / 24
7
+ q/kv heads : 12 / 4 (G=3, hd=128)
8
+ ffn_hidden : 4096 (SwiGLU)
9
+ context_length : 16384
10
+ vocab_size : 50304 tying=1
11
+ B x T : 1 x 16384
12
+ world_size : 2 (dp=1 x cp=2)
13
+ CP_ATTN_MODE : ring (load_balance=HeadTail)
14
+ global_batch : 16384 (grad_accum=1)
15
+ precision : bf16 Linears + fp32 norms/residual/logits (fp32 master weights in AdamW) [mask=15 QKV WO MLP HEAD]
16
+ max_lr/min_lr : 0.0006 / 6e-05
17
+ warmup/steps : 100 / 3 (cosine, no stable phase)
18
+ checkpointing : false (ckpt_freq=200, dir=checkpoints_bluscriptcp)
19
+ =============================================================
20
+ Initializing model on CUDA device 0...
21
+ [YaRN] build_rope_cache: seq_len=16384 hd=128 scale=4.000 low=14 high=32 m=1.1386
22
+ [InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_680m/rank_0/varybatch.ckpt (base step 3400, loss 2.31541) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
23
+ Parameters: 681328128
24
+ Parameters per GPU: 681328128
25
+ max_steps: 3
26
+ warmup_steps: 100
27
+ [ZeRO-1 v2] build=v2.4 supports: rs+ag | can-symmetric-window | can-fp32-reduce | can-bf16-reduce | can-reshard-v1-ckpt
28
+ [ZeRO-1 v2] ACTIVE: reduce_dtype=bfloat16 (= param dtype) symmetric=off overlap=-/-
29
+ [ZeRO-1 v2] zero1-v2 reduce-scatter/all-gather (standard memory)
30
+ [ZeRO-1 v2] params=603979776 padded=603979776 W=2 shard=301989888 | 24 buckets of 50 MB (13107200 elems/rank/bucket) | optimizer state/GPU: 3456 MB (vs 6912 MB unsharded)
31
+ [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.
32
+ [ZeRO-1 v2] build=v2.4 supports: rs+ag | can-symmetric-window | can-fp32-reduce | can-bf16-reduce | can-reshard-v1-ckpt
33
+ [ZeRO-1 v2] ACTIVE: reduce_dtype=float32 (= param dtype) symmetric=off overlap=-/-
34
+ [ZeRO-1 v2] zero1-v2 reduce-scatter/all-gather (standard memory)
35
+ [ZeRO-1 v2] params=77348352 padded=77348352 W=2 shard=38674176 | 6 buckets of 50 MB (6553600 elems/rank/bucket) | optimizer state/GPU: 442.591 MB (vs 885.182 MB unsharded)
36
+ [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.
37
+ [ZeRO-1] variant=v2 reduce-scatter/all-gather groups=2 [bfloat16/603979776] [float32/77348352]
38
+ [ZeRO-1] DataParallel is DISABLED (both would double-reduce)
39
+ [precision] bf16 params=144, all with fp32 master weights
40
+ found 7 shards for split train
41
+ found 1 shards for split val
42
+ Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log11.csv (run 11)
43
+ [rope_theta] 500000 from CP_ROPE_THETA (no sidecar -- provenance UNVERIFIED against the checkpoint)
44
+ Checkpointing: OFF
45
+ [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)
46
+
47
+ Starting training...
48
+
49
+ [CP ring P2P] OVERLAP: dedicated non-blocking stream
50
+ validation loss: 2.8726
51
+ [CP bwd] overlap=ON
52
+ step 0 | loss: 3.535873 | lr 1.8000e-06 | norm: 2.4144 | dt: 2955.96ms | tok/sec: 5542.7 | mem: 45445MB | Time Left: 00 hrs : 00 mins
53
+ [TIMING] data: 0.2ms | fwd: 86.2ms | loss: 0.6ms | bwd: 177.2ms | clip: 0.0ms | optim: 18.7ms
54
+ step 1 | loss: 3.620002 | lr 3.6000e-06 | norm: 2.3326 | dt: 268.35ms | tok/sec: 61054.7 | mem: 50117MB | Time Left: 00 hrs : 00 mins
55
+ [TIMING] data: 0.0ms | fwd: 72.0ms | loss: 0.6ms | bwd: 175.4ms | clip: 0.0ms | optim: 18.3ms
56
+ validation loss: 2.8714
57
+ step 2 | loss: 2.400090 | lr 5.4000e-06 | norm: 1.7171 | dt: 1727.47ms | tok/sec: 9484.4 | mem: 50117MB | Time Left: 00 hrs : 00 mins
58
+ [TIMING] data: 0.0ms | fwd: 81.3ms | loss: 0.6ms | bwd: 175.1ms | clip: 0.0ms | optim: 18.3ms
59
+
60
+ === Training complete ===