Views
No views yet
| metric | value |
|---|---|
step | 17196 |
smooth_train_loss | 2.8574668335763356 |
min_objective | 0.9634662569094967 |
flops_used | 2.8502607518490427e+19 |
flops_per_token | 3161456704.0 |
total_training_time | 58589.97425246239 |
1{
2 "model_pt": {
3 "sequence_len": 1024,
4 "vocab_size": 64000,
5 "n_layer": 24,
6 "n_head": 16,
7 "n_embd": 1024,
8 "intermediate_size": 4096,
9 "rope_theta": 10000.0,
10 "attention_dropout": 0.1,
11 "weight_tying": false,
12 "head_bias": false,
13 "initializer_range": 0.02,
14 "nanochat_init": false,
15 "attn_implementation": "sdpa",
16 "explicit_additive_mask": true
17 },
18 "model_ppt": null,
19 "optim": {
20 "matrix_lr": 0.0005,
21 "embedding_lr": 0.0005,
22 "unembedding_lr": 0.0005,
23 "weight_decay": 0.0001
24 },
25 "train": {
26 "num_iterations": 1000,
27 "eval_every_n_steps": 1000,
28 "eval_every_flops_frac": null,
29 "save_every_n_steps": null,
30 "save_at_steps": null,
31 "log_every_n_steps": 10,
32 "eval_at_end": true,
33 "save_at_end": true,
34 "grad_clip": 1.0,
35 "ema_beta": 0.9
36 },
37 "eval": {
38 "eval_steps": 16,
39 "eval_tokens": 2016000
40 },
41 "hardware": {
42 "device_batch_size": 64,
43 "grad_accum_steps": 4,
44 "peak_tflops": 2250.0
45 },
46 "wandb": {
47 "enabled": true,
48 "notes": "",
49 "group": "repro_nca_owt",
50 "tags": [
51 "hf_llama",
52 "vocab_64000",
53 "pt_owt-gpt2bpe-9B",
54 "seed_0",
55 "ppt_nca-paper-share10-1024",
56 "ppt_lr_trapezoid",
57 "reinit"
58 ],
59 "entity": null
60 },
61 "data_pt": "owt-gpt2bpe-9B",
62 "data_ppt": "nca-paper-share10-1024",
63 "extra_eval": [
64 "fineweb-gpt2bpe-20B",
65 "c4-gpt2bpe-10B"
66 ],
67 "train_split": "train",
68 "pad_vocab": 64000,
69 "lr_kind": "cosine",
70 "lr_warmup_ratio": 0.1,
71 "lr_warmdown_ratio": 0.0,
72 "lr_final_frac": 0.0,
73 "ppt_lr": 0.0001,
74 "ppt_weight_decay": 0.0,
75 "ppt_grad_clip": 0.0,
76 "ppt_device_batch_size": 8,
77 "ppt_grad_accum_steps": 2,
78 "ppt_lr_kind": "trapezoid",
79 "ppt_lr_warmup_ratio": 1.0,
80 "ppt_lr_warmdown_ratio": 0.0,
81 "ppt_tokens": 164160000,
82 "pt_tokens": null,
83 "eval_tokens": 2016000,
84 "reinit_embed_at_transition": true,
85 "reset_optimizer_at_transition": true,
86 "depth": null,
87 "compile_model": false,
88 "model_project": "nca-repro",
89 "run_name": "nca_owt_share10_untied_seed0",
90 "target_flops": 2.9022e+19,
91 "alpha_ppt": 0.0,
92 "use_measured_flops": true,
93 "flops_per_token": null,
94 "seed": 0,
95 "push_to_hf": true,
96 "hf_repo_org": "alexkstern"
97}model_017196.pt — model weights (state_dict).meta_017196.json — training metadata.config_017196.json — run config snapshot.rng_017196.pt — RNG state (when present).