duoduoyeah commited on
Commit
91d2297
·
verified ·
1 Parent(s): 8bd5b18

Add files using upload-large-folder tool

Browse files
base_checkpoints/d4/meta_000416.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "step": 416,
3
+ "val_bpb": 1.137863058912311,
4
+ "model_config": {
5
+ "sequence_len": 1024,
6
+ "pure_vocab_size": 4096,
7
+ "all_vocab_size": 4917,
8
+ "n_layer": 4,
9
+ "n_head": 2,
10
+ "n_kv_head": 2,
11
+ "n_embd": 256,
12
+ "prefix_pure_tokens": 1,
13
+ "mask_token_id": 4096
14
+ },
15
+ "user_config": {
16
+ "run": "pdlm_depth4_bs8_pr1_ratio40_causal",
17
+ "device_type": "",
18
+ "depth": 4,
19
+ "max_seq_len": 1024,
20
+ "block_size": 8,
21
+ "prefix_pure_tokens": 1,
22
+ "is_causal": true,
23
+ "num_iterations": -1,
24
+ "target_flops": -1.0,
25
+ "target_param_data_ratio": 40,
26
+ "device_batch_size": 64,
27
+ "total_batch_size": 524288,
28
+ "embedding_lr": 0.2,
29
+ "unembedding_lr": 0.004,
30
+ "weight_decay": 0.0,
31
+ "matrix_lr": 0.02,
32
+ "grad_clip": 1.0,
33
+ "warmup_ratio": 0.0,
34
+ "warmdown_ratio": 0.2,
35
+ "final_lr_frac": 0.0,
36
+ "resume_from_step": -1,
37
+ "eval_every": 99999,
38
+ "eval_tokens": 10485760,
39
+ "core_metric_every": -1,
40
+ "core_metric_max_per_task": 500,
41
+ "sample_every": 2000,
42
+ "save_every": -1,
43
+ "model_tag": ""
44
+ },
45
+ "device_batch_size": 64,
46
+ "max_seq_len": 1024,
47
+ "dataloader_state_dict": {
48
+ "pq_idx": 3,
49
+ "rg_idx": 129
50
+ },
51
+ "loop_state": {
52
+ "min_val_bpb": 1.137863058912311,
53
+ "smooth_train_loss": 2.1223166214609748,
54
+ "total_training_time": 308.1995761394501
55
+ }
56
+ }
base_checkpoints/d4/model_000416.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64aa68c5df3e765120c62eee2e662dfd152fd361377d8656bd1c994153a2d933
3
+ size 19305717
base_checkpoints/d4/optim_000416_rank0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa59aeb2c2c48281a6026592c20badb8b278d7978eb64defb7876092984676af
3
+ size 26017109
report/base-model-training.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## Base model training
2
+ timestamp: 2025-12-27 20:59:08
3
+
4
+ - run: pdlm_depth4_bs8_pr1_ratio40_causal
5
+ - device_type:
6
+ - depth: 4
7
+ - max_seq_len: 1024
8
+ - block_size: 8
9
+ - prefix_pure_tokens: 1
10
+ - is_causal: True
11
+ - num_iterations: -1
12
+ - target_flops: -1.0000
13
+ - target_param_data_ratio: 40
14
+ - device_batch_size: 64
15
+ - total_batch_size: 524,288
16
+ - embedding_lr: 0.2000
17
+ - unembedding_lr: 0.0040
18
+ - weight_decay: 0.0000
19
+ - matrix_lr: 0.0200
20
+ - grad_clip: 1.0000
21
+ - warmup_ratio: 0.0000
22
+ - warmdown_ratio: 0.2000
23
+ - final_lr_frac: 0.0000
24
+ - resume_from_step: -1
25
+ - eval_every: 99,999
26
+ - eval_tokens: 10,485,760
27
+ - core_metric_every: -1
28
+ - core_metric_max_per_task: 500
29
+ - sample_every: 2000
30
+ - save_every: -1
31
+ - model_tag:
32
+ - Number of parameters: 5,453,056
33
+ - Number of FLOPs per token: 3.774874e+07
34
+ - Calculated number of iterations: 416
35
+ - Number of training tokens: 218,103,808
36
+ - Tokens : Params ratio: 39.9966
37
+ - DDP world size: 1
38
+ - warmup_ratio: 0.0000
39
+ - warmdown_ratio: 0.2000
40
+ - final_lr_frac: 0.0000
41
+ - Minimum validation bpb: 1.1379
42
+ - Final validation bpb: 1.1379
43
+ - CORE metric estimate: None
44
+ - MFU %: 2.64%
45
+ - Total training flops: 8.233143e+15
46
+ - Total training time: 5.14m
47
+ - Peak memory usage: 6905.37MiB
48
+
tokenizer/token_bytes.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c8b99c0d5a1b87b87118e840f69510440302023cd514b241614fb562373d7ce
3
+ size 17961
tokenizer/token_maps.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e018730860c8e77e8cbba4b57ac9c7ba6798b5926dce925743959b932a099964
3
+ size 1850237
tokenizer/tokenizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f874c4250ec76e2e8c4f97e91c55cfdf74d9f8eedaae14cd22db36bb718ee19
3
+ size 61662