rubentium commited on
Commit
3447109
·
verified ·
1 Parent(s): 5e889ec

Upload seed-72 full resume checkpoint

Browse files
FINAL5_20261014/seed72_dense/checkpoint-20000/config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "vocab_size": 151936,
3
+ "hidden_size": 1536,
4
+ "intermediate_size": 6144,
5
+ "num_hidden_layers": 28,
6
+ "num_attention_heads": 12,
7
+ "num_key_value_heads": 4,
8
+ "head_dim": 128,
9
+ "hidden_act": "silu",
10
+ "max_position_embeddings": 32768,
11
+ "initializer_range": 0.02,
12
+ "rms_norm_eps": 1e-06,
13
+ "use_cache": true,
14
+ "tie_word_embeddings": true,
15
+ "rope_theta": 1000000,
16
+ "rope_scaling": null,
17
+ "positional": "rope",
18
+ "qk_norm": "on",
19
+ "qk_gain_fp32": true,
20
+ "norm_gain_fp32": false,
21
+ "qk_gain_mode": "vector",
22
+ "rope_train_position_scale_set": null,
23
+ "attention_bias": false,
24
+ "attention_dropout": 0.0,
25
+ "bos_token_id": 151643,
26
+ "eos_token_id": 151643,
27
+ "checkpoint_activations": false,
28
+ "attention_type": "dense",
29
+ "sparse_block_size": 128,
30
+ "sparse_num_random_blocks": 31,
31
+ "sparse_mask_pool_size": 1,
32
+ "sparse_max_seq_length": 8192,
33
+ "sparse_layer_indices": null,
34
+ "sparse_mask_cache_dir": null,
35
+ "sparse_mask_seed": 0,
36
+ "sparse_mixed_nrb_set": null,
37
+ "moba_block_size": 128,
38
+ "moba_top_k": 32,
39
+ "asentmax_alpha": 1.5,
40
+ "asentmax_learn_temp": true,
41
+ "asentmax_init_temp": 1.0
42
+ }
FINAL5_20261014/seed72_dense/checkpoint-20000/loader_gen_state_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21b4160791b11c535b1414390199117f5857a94ac58d7b970468583a72bb0bc
3
+ size 6802
FINAL5_20261014/seed72_dense/checkpoint-20000/loader_gen_state_rank1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32ec129fe94d28aa0ddb1827422f7723c72b31ebc7a9306e98894b6131bb7fe9
3
+ size 6802
FINAL5_20261014/seed72_dense/checkpoint-20000/loader_gen_state_rank2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c5588d56d5c8519bb2c45d4b62d3684da8142cd70467b7c6b989cd944f4e533
3
+ size 6802
FINAL5_20261014/seed72_dense/checkpoint-20000/loader_gen_state_rank3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05faccfef6d5dfcc8dd8aca9b50fb4a4f1b5e5ee52a9a29e99e1e68bd3e62b77
3
+ size 6802
FINAL5_20261014/seed72_dense/checkpoint-20000/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d940e694bf6f47dd65d802a64c48dcd50a91f13bb139fcf1375937fbaffb7eb4
3
+ size 2404833883
FINAL5_20261014/seed72_dense/checkpoint-20000/rng_state_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa8f3f2527a73f03263b1714be4aa8be5be0e5644561efb31e37452dcb230636
3
+ size 14821
FINAL5_20261014/seed72_dense/checkpoint-20000/rng_state_rank1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:892527a26e05f13ce99dd25d7deecef48be5561b60d5cb549df9385df1704cb9
3
+ size 14885
FINAL5_20261014/seed72_dense/checkpoint-20000/rng_state_rank2.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50617cc1166cd88f7dc063da3765c88e8b238cc8a34ab7aeeaa7be12a7296e61
3
+ size 14821
FINAL5_20261014/seed72_dense/checkpoint-20000/rng_state_rank3.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ff4c836741720716968802bd438c08888f256ab52b5e21f2897ee8d810f75de
3
+ size 14885
FINAL5_20261014/seed72_dense/checkpoint-20000/train_config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_path": "../qwen3_06b_base",
3
+ "model_dtype": "bfloat16",
4
+ "init_from_scratch": true,
5
+ "train_path": "../../data/fineweb_large/tokenized",
6
+ "val_path": null,
7
+ "max_seq_length": 8192,
8
+ "num_workers": 8,
9
+ "data_seed": 72,
10
+ "concatenate_seeded_8k": false,
11
+ "output_dir": "models/FINAL5_20261014/1p2b_dense_qkgainfp32_s72/run_20261014_191331",
12
+ "resume_from_checkpoint": "models/FINAL5_20261014/1p2b_dense_qkgainfp32_s72/run_20261014_191331/checkpoint-2000",
13
+ "force_new_run": false,
14
+ "num_epochs": 1,
15
+ "max_steps": 50462,
16
+ "per_device_batch_size": 1,
17
+ "gradient_accumulation_steps": 15,
18
+ "learning_rate": 0.0005,
19
+ "weight_decay": 0.01,
20
+ "warmup_steps": 750,
21
+ "lr_scheduler": "cosine",
22
+ "max_grad_norm": 1.0,
23
+ "save_steps": 5000,
24
+ "eval_steps": 250,
25
+ "eval_batches": 50,
26
+ "logging_steps": 10,
27
+ "seed": 72,
28
+ "verbose": true,
29
+ "checkpoint_activations": false,
30
+ "resume_micro_step_override": 30000,
31
+ "strategy": "ddp",
32
+ "mixed_precision": "bf16",
33
+ "wandb_enabled": false,
34
+ "wandb_project": "stochastic_sparsity",
35
+ "wandb_name": "FINAL5 1.2B Dense seed 72",
36
+ "wandb_tags": [
37
+ "FINAL5",
38
+ "1p2b",
39
+ "dense",
40
+ "seed72",
41
+ "qk_gain_fp32"
42
+ ],
43
+ "hidden_size": 1536,
44
+ "intermediate_size": 6144,
45
+ "num_attention_heads": 12,
46
+ "num_key_value_heads": 4,
47
+ "num_hidden_layers": 28,
48
+ "head_dim": 128,
49
+ "attention_dropout": 0.0,
50
+ "rope_theta": 1000000,
51
+ "rope_train_position_scale_set": null,
52
+ "positional": "rope",
53
+ "qk_norm": "on",
54
+ "qk_gain_fp32": true,
55
+ "norm_gain_fp32": false,
56
+ "qk_gain_mode": "vector",
57
+ "attention_type": "dense",
58
+ "sparse_block_size": 128,
59
+ "sparse_num_random_blocks": 31,
60
+ "sparse_mask_pool_size": 1,
61
+ "sparse_val_dense_compare": false,
62
+ "sparse_layer_indices": null,
63
+ "sparse_mask_cache_dir": null,
64
+ "sparse_mask_seed": 0,
65
+ "sparse_mixed_nrb_set": null,
66
+ "dense_mode": false,
67
+ "dense_step_intervals": null
68
+ }
FINAL5_20261014/seed72_dense/checkpoint-20000/trainer_state.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "global_step": 20000,
3
+ "epoch": 0,
4
+ "micro_step": 300000,
5
+ "sampler_seed": 72,
6
+ "seed": 72,
7
+ "world_size": 4,
8
+ "optimizer_state_file": "training_state.pt"
9
+ }
FINAL5_20261014/seed72_dense/checkpoint-20000/training_state.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4eb34c63137098695159f5ff02d832f15b07a555af95406f407338020b89428
3
+ size 4809712479