TightX commited on
Commit
4e79b6e
·
verified ·
1 Parent(s): d2020f5

add arch_facts.json

Browse files
Files changed (1) hide show
  1. arch_facts.json +59 -0
arch_facts.json ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "params_total": 1095845888,
3
+ "n_tensors": 143,
4
+ "dtype": "bf16",
5
+ "vocab_size": 32000,
6
+ "d_model": 2048,
7
+ "n_blocks": 20,
8
+ "tensor_keys_sample": [
9
+ "blocks.0.attn.out_proj.weight",
10
+ "blocks.0.attn.qkv.weight",
11
+ "blocks.0.attn_norm.scale",
12
+ "blocks.0.ffn.down.weight",
13
+ "blocks.0.ffn.gate.weight",
14
+ "blocks.0.ffn.up.weight",
15
+ "blocks.0.ffn_norm.scale",
16
+ "blocks.1.attn.out_proj.weight"
17
+ ],
18
+ "source_checkpoint": "<workspace>/1B WSM/fine-tuning/ckpt_1b_mix42_merged/best_val/checkpoint.pt",
19
+ "source_checkpoint_relative": "fine-tuning/ckpt_1b_mix42_merged/best_val/checkpoint.pt",
20
+ "source_keys": [
21
+ "model",
22
+ "optimizer",
23
+ "scheduler",
24
+ "config",
25
+ "training_state",
26
+ "timestamp"
27
+ ],
28
+ "source_config": {
29
+ "vocab_size": 32000,
30
+ "d_model": 2048,
31
+ "num_layers": 20,
32
+ "num_heads": 16,
33
+ "d_ff": 5120,
34
+ "max_seq_len": 1024,
35
+ "rope_theta": 500000.0,
36
+ "dropout_rate": 0.0
37
+ },
38
+ "config_note": "use_fp8 absent from the raw checkpoint config (plain nn.Linear build); none was added.",
39
+ "checkpoint_step": 2000,
40
+ "checkpoint_timestamp": "2026-10-13T23:27:36.964113",
41
+ "training_state": {
42
+ "step": 2000,
43
+ "total_tokens": 65536000,
44
+ "epoch": 0,
45
+ "best_val_loss": 0.6519121726595375,
46
+ "max_tokens": 31013881856,
47
+ "round_start_tokens": 0,
48
+ "data_position": {
49
+ "source_type": "bin",
50
+ "file_index": 0,
51
+ "files_total": 0,
52
+ "rows_skipped": 0,
53
+ "bin_path": "data_mix_gold/sft_train.bin",
54
+ "token_offset": 68812800
55
+ }
56
+ },
57
+ "nonfinite_after_cast": [],
58
+ "created_utc": "2026-10-23T11:01:58Z"
59
+ }