rubentium commited on
Commit
8e387c0
·
verified ·
1 Parent(s): bcd5e3e

Add FINAL5 seed72 Mixed step27000 resume manifest and config

Browse files
checkpoints/FINAL5_20261014/part_t/1p2b_mixed_qkgainfp32_s72/checkpoint-27000-rcp/RESUME_MANIFEST.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # FINAL5 seed 72 Mixed checkpoint 27000 resume bundle
2
+
3
+ Captured on 2026-10-28. This is an intermediate resume checkpoint at step 27,000 of 50,462, not a final checkpoint. Do not evaluate it as a completed seed pair.
4
+
5
+ The archived checkpoint includes model weights, optimizer and scheduler state, the saved training configuration, loader state, and RNG state. It was verified after upload against the source file sizes and SHA-256/LFS hashes (10 of 10 files passed; 7,214,606,594 bytes).
6
+
7
+ Resume settings: seed 72, qk_gain_fp32=true, maximum step 50,462, global batch 60, per-device batch 1, gradient accumulation 20, and checkpoint cadence 1,000 steps. The saved scheduler LR at step 27,000 was preserved on resume. The three-rank DDP resume used resume_micro_step_override=540000 to restore the epoch-start loader RNG and replay the expected shuffled order.
8
+
9
+ Training code revision: 9a5045f08a84d089699885e3685fc55d8feb7060; tracked working-tree changes were absent. The exact resume configuration is in ../resume_config_3gpu_global_batch60.json.
10
+
11
+ The Mixed seed 72 mask cache is already present in this repository at FINAL5_20261014/mask_cache_8k_mixed_seed72/ (manifest plus 28 layer files). The FineWeb-Edu training export and tokenizer are not included in this checkpoint directory and must be available at the paths specified by the resume configuration.
12
+
13
+ | File | Bytes | SHA-256 |
14
+ |---|---:|---|
15
+ | config.json | 1,153 | 19f448cd58000f7c4c1e368ac043e43e5ac9612188dd2c5df01719255ecabac6 |
16
+ | loader_epoch_start_state_rank0.pt | 6,858 | 0656ddbff730b3ef739cdb4d980f54ff169c931544ce7a093c21a7dc6fa6f49e |
17
+ | loader_epoch_start_state_rank1.pt | 6,858 | c8f9144eba7ef521ca55f126341e4d11d4584a2c550212432b63364585392f0d |
18
+ | loader_gen_state_rank0.pt | 6,802 | b21b4160791b11c535b1414390199117f5857a94ac58d7b970468583a72bb0bc |
19
+ | loader_gen_state_rank1.pt | 6,802 | 32ec129fe94d28aa0ddb1827422f7723c72b31ebc7a9306e98894b6131bb7fe9 |
20
+ | model.pt | 2,404,833,883 | cbddfa7b1fc496462310e94c0140a213b7424d61a29ac7ed1ea37436cd0a8db2 |
21
+ | rng_state_rank0.pt | 14,821 | 60e53db818a76f083feffaac06d1144bc0f16781bd3c027e99c826b473d848e7 |
22
+ | rng_state_rank1.pt | 14,885 | 3c6a920ab76f3c9355035b7fe950d73e6efa67247394e39a1d9f7f46283e0842 |
23
+ | training_config.json | 1,989 | b6a80a40ea56e203679ec84b0e3ab5f3a09d27e390c409ee70ab1eaa6edc75fd |
24
+ | training_state.pt | 4,809,712,543 | 985fc0f558fcb6736ffe50c0e0e5d687dc50c3dd7501bb4f9c54a970374e768e |
25
+
26
+ The checkpoint was resumed after this archive was made; the 27,000-step checkpoint itself was not evaluated.
checkpoints/FINAL5_20261014/part_t/1p2b_mixed_qkgainfp32_s72/resume_config_3gpu_global_batch60.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "path": "../../qwen3_06b_base",
4
+ "dtype": "bfloat16",
5
+ "init_from_scratch": true,
6
+ "hidden_size": 1536,
7
+ "intermediate_size": 6144,
8
+ "num_attention_heads": 12,
9
+ "num_key_value_heads": 4,
10
+ "qk_gain_fp32": true
11
+ },
12
+ "data": {
13
+ "train_path": "../../data/fineweb_large/tokenized",
14
+ "val_path": null,
15
+ "max_seq_length": 8192,
16
+ "num_workers": 0,
17
+ "seed": 72
18
+ },
19
+ "training": {
20
+ "output_dir": "models/FINAL5_20261014/1p2b_mixed_qkgainfp32_s72",
21
+ "resume_from_checkpoint": "models/FINAL5_20261014/1p2b_mixed_qkgainfp32_s72/run_20261015_193033/checkpoint-27000",
22
+ "force_new_run": false,
23
+ "num_epochs": 1,
24
+ "max_steps": 50462,
25
+ "per_device_batch_size": 1,
26
+ "gradient_accumulation_steps": 20,
27
+ "learning_rate": 0.0005,
28
+ "weight_decay": 0.01,
29
+ "warmup_steps": 750,
30
+ "lr_scheduler": "cosine",
31
+ "max_grad_norm": 1.0,
32
+ "save_steps": 1000,
33
+ "eval_steps": 250,
34
+ "eval_batches": 50,
35
+ "logging_steps": 10,
36
+ "seed": 72,
37
+ "verbose": true,
38
+ "checkpoint_activations": false,
39
+ "global_batch_size": 60,
40
+ "keep_last_checkpoints": 3,
41
+ "keep_every_n_steps": 10000,
42
+ "resume_micro_step_override": 540000
43
+ },
44
+ "distributed": {
45
+ "strategy": "ddp",
46
+ "mixed_precision": "bf16"
47
+ },
48
+ "wandb": {
49
+ "enabled": false,
50
+ "project": "stochastic_sparsity",
51
+ "name": "FINAL5_20261014 1p2b_mixed_qkgainfp32_s72",
52
+ "tags": [
53
+ "qwen3",
54
+ "1.2b",
55
+ "8k",
56
+ "p2",
57
+ "mixed",
58
+ "seed72"
59
+ ]
60
+ },
61
+ "attention": {
62
+ "type": "sparse_stochastic",
63
+ "dense_mode": false,
64
+ "sparse_block_size": 128,
65
+ "sparse_num_random_blocks": 35,
66
+ "sparse_mask_pool_size": 10000,
67
+ "sparse_mixed_nrb_set": [
68
+ 63,
69
+ 47,
70
+ 31,
71
+ 21,
72
+ 11
73
+ ],
74
+ "sparse_val_dense_compare": false,
75
+ "dense_step_intervals": null,
76
+ "sparse_layer_indices": null,
77
+ "sparse_mask_cache_dir": "../../sparse/mask_cache_8k_mixed_seed72",
78
+ "sparse_mask_seed": 72
79
+ }
80
+ }