Views
No views yet
| metric | value |
|---|---|
step | 370 |
smooth_train_loss | 5.399075984954834 |
min_objective | 7.79401826513968 |
flops_used | 4.493192062618829e+17 |
flops_per_token | 2316238848.0 |
total_training_time | 297.2031021118164 |
1{
2 "model_pt": {
3 "sequence_len": 2048,
4 "vocab_size": 65536,
5 "n_layer": 18,
6 "n_head": 9,
7 "n_kv_head": 9,
8 "n_embd": 1152
9 },
10 "model_ppt": {
11 "sequence_len": 2048,
12 "vocab_size": 256,
13 "n_layer": 18,
14 "n_head": 9,
15 "n_kv_head": 9,
16 "n_embd": 1152
17 },
18 "optim": {
19 "matrix_lr": 3e-05,
20 "embedding_lr": 3e-05,
21 "unembedding_lr": 3e-05,
22 "weight_decay": 0.0
23 },
24 "train": {
25 "num_iterations": 1000,
26 "eval_every_n_steps": null,
27 "eval_every_flops_frac": null,
28 "save_every_n_steps": null,
29 "save_at_steps": null,
30 "log_every_n_steps": 100,
31 "eval_at_end": true,
32 "save_at_end": true,
33 "grad_clip": 1.0,
34 "ema_beta": 0.0
35 },
36 "eval": {
37 "eval_steps": 16,
38 "eval_tokens": 2016000
39 },
40 "hardware": {
41 "device_batch_size": 32,
42 "grad_accum_steps": 4,
43 "peak_tflops": 2250.0
44 },
45 "wandb": {
46 "enabled": true,
47 "notes": "",
48 "group": "9e18_ppt0.05",
49 "tags": [
50 "nanochat_gpt",
51 "pt_fineweb-nanochatbpe-20B",
52 "ppt_dyck-k128-seq_len_2048-1B",
53 "seed_2",
54 "case_c",
55 "lr_trapezoid",
56 "depth_18"
57 ],
58 "entity": null
59 },
60 "data_pt": "fineweb-nanochatbpe-20B",
61 "data_ppt": "dyck-k128-seq_len_2048-1B",
62 "extra_eval": [
63 "c4-nanochatbpe-10B"
64 ],
65 "train_split": "train",
66 "pad_vocab": 65536,
67 "ppt_vocab_size": null,
68 "ppt_same_vocab_as_pt": false,
69 "ppt_unified_lr": false,
70 "lr_kind": "trapezoid",
71 "lr_warmup_ratio": 0.0,
72 "lr_warmdown_ratio": 0.4,
73 "lr_final_frac": 0.0,
74 "ppt_lr_kind": "trapezoid",
75 "ppt_lr_warmup_ratio": 0.0,
76 "ppt_lr_warmdown_ratio": 0.4,
77 "ppt_lr_final_frac": 0.0,
78 "ppt_lr": 3e-05,
79 "ppt_weight_decay": 0.0,
80 "ppt_device_batch_size": 32,
81 "ppt_grad_accum_steps": 4,
82 "eval_tokens": 2016000,
83 "reinit_embed_at_transition": true,
84 "moment_match_embed_reinit": false,
85 "reset_optimizer_at_transition": true,
86 "random_pt_head": false,
87 "depth": 18,
88 "compile_model": true,
89 "model_project": "eval_seed_fleet_v1",
90 "run_name": "kdyck_5pct_9e18_d18_seed2",
91 "target_flops": 9e+18,
92 "alpha_ppt": 0.05,
93 "use_measured_flops": true,
94 "flops_per_token": null,
95 "ppt_flops_per_token": null,
96 "seed": 2,
97 "metrics_output_file": null,
98 "push_to_hf": true,
99 "hf_repo_org": "alexkstern",
100 "cleanup_checkpoints": true
101}model_000370.pt — model weights (state_dict).meta_000370.json — training metadata.config_000370.json — run config snapshot.rng_000370.pt — RNG state (when present).