Subject-Emu-5259 commited on
Commit
ae12ce1
Β·
verified Β·
1 Parent(s): 5326aee

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +7 -2
README.md CHANGED
@@ -59,10 +59,15 @@ pipeline_tag: text-generation
59
 
60
  ---
61
 
62
- ## πŸ“ˆ Training (v1 lineage, preserved)
 
 
 
 
 
63
 
64
  <p align="center">
65
- <img src="neuralai-nae1-training.png" alt="Nae1 training loss step 28000 to 32000" width="92%" />
66
  </p>
67
 
68
  Nae1 v1 was pretrained from random initialization in eight cosine cycles: 0 β†’ 4,000 Β· 4,000 β†’ 8,000 Β· … Β· 24,000 β†’ 28,000 Β· 28,000 β†’ 32,000, each resumed from the previous checkpoint. The chart shows cycle eight β€” the **first cycle trained with corrected next-token labels** (an earlier double-shift bug had models predicting token t+2; the held-out gate was always standard next-token, so historical PPL numbers remain valid). Under the corrected objective the logged loss resumes at 28.87, falls to a best of **15.72** (step 31,500) and finishes at **17.93** as the learning rate anneals to zero. Held-out PPL improves 147.97 β†’ 146.17 β†’ **54.16** β€” a 63% drop from fixing the training objective; a fresh-corpus cross-check (unseen FineWeb CC-MAIN-2014-10: **57.09** vs step-28000's **143.64**) confirms the jump is genuine generalization, not eval-set leakage.
 
59
 
60
  ---
61
 
62
+ ## πŸ“ˆ Training
63
+
64
+ **Current release (v2).** In-kernel held-out PPL, steps 62000 β†’ 80000 (bf16, 200k-token slice): 47.11 β†’ **43.09**, decelerating into the cosine floor β€” the reason the next run re-warms to peak LR 6e-4 instead of adding more low-LR steps.
65
+
66
+
67
+ ### v1 lineage (frozen)
68
 
69
  <p align="center">
70
+ <img src="neuralai-nae1-training.png" alt="Nae1 v1 training loss step 28000 to 32000" width="92%" />
71
  </p>
72
 
73
  Nae1 v1 was pretrained from random initialization in eight cosine cycles: 0 β†’ 4,000 Β· 4,000 β†’ 8,000 Β· … Β· 24,000 β†’ 28,000 Β· 28,000 β†’ 32,000, each resumed from the previous checkpoint. The chart shows cycle eight β€” the **first cycle trained with corrected next-token labels** (an earlier double-shift bug had models predicting token t+2; the held-out gate was always standard next-token, so historical PPL numbers remain valid). Under the corrected objective the logged loss resumes at 28.87, falls to a best of **15.72** (step 31,500) and finishes at **17.93** as the learning rate anneals to zero. Held-out PPL improves 147.97 β†’ 146.17 β†’ **54.16** β€” a 63% drop from fixing the training objective; a fresh-corpus cross-check (unseen FineWeb CC-MAIN-2014-10: **57.09** vs step-28000's **143.64**) confirms the jump is genuine generalization, not eval-set leakage.