Upload logs/t32_probe.log with huggingface_hub
Browse files- logs/t32_probe.log +61 -0
logs/t32_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 |
+
================ 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 : 32768
|
| 10 |
+
vocab_size : 50304 tying=1
|
| 11 |
+
B x T : 1 x 32768
|
| 12 |
+
world_size : 2 (dp=1 x cp=2)
|
| 13 |
+
CP_ATTN_MODE : ring (load_balance=HeadTail)
|
| 14 |
+
global_batch : 524288 (grad_accum=16)
|
| 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][warn] seq_len(32768) != scale(32.000)*orig_maxpos(4096.0)=131072.0 -- cache scaled for a different context than it is sized for
|
| 22 |
+
[YaRN] build_rope_cache: seq_len=32768 hd=128 scale=32.000 low=14 high=32 m=1.3466
|
| 23 |
+
[InitFrom] weights loaded from /root/Adhi/BluTrain/dist/Context_Parallelism/checkpoints_bluscriptcp/rank_0/blumodelcp_run12_step_1200.ckpt (base step 1200, loss 2.68627) -- optimizer state and RNG intentionally NOT restored; pair with CP_REWARMUP
|
| 24 |
+
Parameters: 681328128
|
| 25 |
+
Parameters per GPU: 681328128
|
| 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=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)
|
| 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=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)
|
| 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/603979776] [float32/77348352]
|
| 39 |
+
[ZeRO-1] DataParallel is DISABLED (both would double-reduce)
|
| 40 |
+
[precision] bf16 params=144, all with fp32 master weights
|
| 41 |
+
found 7 shards for split train
|
| 42 |
+
found 1 shards for split val
|
| 43 |
+
Saving logs to: CP_BluScreipt_Training_logs/CP_Training_log58.csv (run 58)
|
| 44 |
+
[rope_theta] 500000 from CP_ROPE_THETA (no sidecar -- provenance UNVERIFIED against the checkpoint)
|
| 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: 2.8864
|
| 52 |
+
[CP bwd] overlap=ON
|
| 53 |
+
step 0 | loss: 2.859253 | lr 1.8000e-06 | norm: 0.6029 | dt: 15011.20ms | tok/sec: 34926.4 | mem: 78517MB | Time Left: 00 hrs : 00 mins
|
| 54 |
+
[TIMING] data: 6.8ms | fwd: 2661.5ms | loss: 17.7ms | bwd: 7690.7ms | clip: 0.0ms | optim: 18.9ms
|
| 55 |
+
step 1 | loss: 2.673292 | lr 3.6000e-06 | norm: 0.5749 | dt: 10351.92ms | tok/sec: 50646.5 | mem: 77525MB | Time Left: 00 hrs : 00 mins
|
| 56 |
+
[TIMING] data: 6.4ms | fwd: 2648.5ms | loss: 17.7ms | bwd: 7650.6ms | clip: 0.0ms | optim: 18.4ms
|
| 57 |
+
validation loss: 2.8855
|
| 58 |
+
step 2 | loss: 2.624726 | lr 5.4000e-06 | norm: 0.6049 | dt: 13763.30ms | tok/sec: 38093.2 | mem: 77397MB | Time Left: 00 hrs : 00 mins
|
| 59 |
+
[TIMING] data: 6.4ms | fwd: 2661.3ms | loss: 17.8ms | bwd: 7673.4ms | clip: 0.0ms | optim: 18.6ms
|
| 60 |
+
|
| 61 |
+
=== Training complete ===
|