SURELYNOTVAIN commited on
Commit
2c2814f
·
verified ·
1 Parent(s): 97f1dda

Table-only prerenders, progress table image, no duplicated tables

Browse files
.gitattributes CHANGED
@@ -38,3 +38,4 @@ charts/average_table.png filter=lfs diff=lfs merge=lfs -text
38
  charts/avg_7tasks.png filter=lfs diff=lfs merge=lfs -text
39
  charts/benchmark_table.png filter=lfs diff=lfs merge=lfs -text
40
  charts/compute_table.png filter=lfs diff=lfs merge=lfs -text
 
 
38
  charts/avg_7tasks.png filter=lfs diff=lfs merge=lfs -text
39
  charts/benchmark_table.png filter=lfs diff=lfs merge=lfs -text
40
  charts/compute_table.png filter=lfs diff=lfs merge=lfs -text
41
+ charts/progress_table.png filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -63,13 +63,13 @@ Author: <b>Dominik Andrzejewski</b>
63
 
64
  ## Training lineage
65
 
66
- | # | Stage | Tokens | What happened |
67
- |---|---|---:|---|
68
- | 1 | **Pretraining (75M)** | 4.0B | A 75M-parameter model was pretrained from scratch on English text, single pass. |
69
- | 2 | **Growth to 100M** | 0.61B | Grown from 12 to 18 layers (identity-initialised) and pretrained further on new documents. |
70
- | 3 | **Chat fine-tune** | 0.18B | Fine-tuned for chat. These weights are the starting point of the expansion. |
71
- | 4 | **Expansion** | about 3.3B | Continued pretraining on new documents from the same sources, to step 15,000 with a 262k-token batch, on a cloud A100 in two chunks. |
72
- | 5 | **Cooldown** | 1.05B | The learning rate was decayed to its minimum over 4,000 steps on a second, unseen corpus, starting from step 15,000. **This is the checkpoint released here.** |
73
 
74
  A separate local cooldown (170 steps from step 13,000) was an experiment. It is **not** in this release's lineage.
75
 
@@ -109,7 +109,7 @@ The 75M pretraining (stage 1) used the same sources with more FineWeb-Edu (37%)
109
 
110
  Chat data was filtered: refusals, boilerplate, other models' identities and over-length chats were dropped, and any sexual content involving minors was removed.
111
 
112
- ### Evaluation
113
 
114
  | Benchmark | Dataset |
115
  |---|---|
@@ -158,14 +158,9 @@ The highlighted cell in each column is the best score in that column, whichever
158
 
159
  ### Progress across checkpoints of this model
160
 
161
- | Checkpoint | Val loss | arc_challenge | arc_easy | hellaswag | openbookqa | piqa | sciq | winogrande | Avg (7) | BLiMP |
162
- |---|---:|---:|---:|---:|---:|---:|---:|---:|---:|---:|
163
- | Expansion step 9,500 | 2.7780 | 27.6 | 52.8 | **29.8** | 27.6 | 62.4 | 71.0 | 51.6 | 46.11 | 76.88 |
164
- | Expansion step 13,000 | 2.7722 | 28.8 | 49.8 | 28.4 | 29.0 | 61.6 | 70.4 | 51.8 | 45.69 | 78.03 |
165
- | Expansion step 15,000 (cooldown start) | 2.7730 | 26.8 | 52.6 | 29.0 | **30.0** | 62.0 | 69.8 | 53.0 | 46.17 | 77.66 |
166
- | **Cooldown step 4,000 (this release)** | **2.6531** | **29.2** | **54.0** | 29.4 | 29.4 | **62.8** | **72.2** | **54.0** | **47.29** | **78.70** |
167
 
168
- Bold marks the best value in each column (lowest for val loss). Validation loss is measured on the held-out split of the first expansion corpus, which the model trained on. It is not a clean test, so the cooldown's lower loss should not be read as general improvement on its own. The benchmark average rose after the cooldown and is the more useful signal.
169
 
170
  ## Compute
171
 
 
63
 
64
  ## Training lineage
65
 
66
+ 1. **Pretraining (75M):** a 75M-parameter model was pretrained from scratch on English text, single pass.
67
+ 2. **Growth to 100M:** grown from 12 to 18 layers (identity-initialised) and pretrained further on new documents.
68
+ 3. **Chat fine-tune:** fine-tuned for chat. These weights are the starting point of the expansion.
69
+ 4. **Expansion:** continued pretraining on new documents from the same sources, to step 15,000 with a 262k-token batch.
70
+ 5. **Cooldown:** the learning rate was decayed to its minimum over 4,000 steps on a second, unseen corpus, starting from step 15,000. **This is the checkpoint released here.**
71
+
72
+ Tokens and hardware for each stage are in [Compute](#compute).
73
 
74
  A separate local cooldown (170 steps from step 13,000) was an experiment. It is **not** in this release's lineage.
75
 
 
109
 
110
  Chat data was filtered: refusals, boilerplate, other models' identities and over-length chats were dropped, and any sexual content involving minors was removed.
111
 
112
+ ### Evaluation benchmarks
113
 
114
  | Benchmark | Dataset |
115
  |---|---|
 
158
 
159
  ### Progress across checkpoints of this model
160
 
161
+ ![Progress across checkpoints](charts/progress_table.png)
 
 
 
 
 
162
 
163
+ The highlighted cell in each column is the best value (lowest for val loss). Validation loss is measured on the held-out split of the first expansion corpus, which the model trained on. It is not a clean test, so the cooldown's lower loss should not be read as general improvement on its own. The benchmark average rose after the cooldown and is the more useful signal.
164
 
165
  ## Compute
166
 
charts/average_table.png CHANGED

Git LFS Details

  • SHA256: fe8414c3c1c1b7af8aea17dfe6c54ce530461655aa746714f09b8709b3771500
  • Pointer size: 131 Bytes
  • Size of remote file: 102 kB

Git LFS Details

  • SHA256: 86ee7519a507b266728cf01ce77123d62131aafb50bcd7be2a9e683319747042
  • Pointer size: 130 Bytes
  • Size of remote file: 57.6 kB
charts/benchmark_table.png CHANGED

Git LFS Details

  • SHA256: 85ea73a9d3694e932433ba5440ee09568f5e0087ed9c74aac3e0f0682b581ba1
  • Pointer size: 131 Bytes
  • Size of remote file: 173 kB

Git LFS Details

  • SHA256: 23995be31daaf877e16deabd214027a07591ae0439b6cdd182a6124ab4ba0a2c
  • Pointer size: 131 Bytes
  • Size of remote file: 147 kB
charts/compute_table.png CHANGED

Git LFS Details

  • SHA256: 033b27aa429b05dd60b15ddebd1fb62335c67a244fb68e137c735fd41f6f477e
  • Pointer size: 131 Bytes
  • Size of remote file: 176 kB

Git LFS Details

  • SHA256: e736d6c1acbc9d3a21dd918b1b277b24ad64c2d8b53ed7ef8cbb0949af7d4d58
  • Pointer size: 131 Bytes
  • Size of remote file: 154 kB
charts/progress_table.png ADDED

Git LFS Details

  • SHA256: 12aee0b7fc9d7ffe10faa8b64e6cf3641d1b4efbc477ddc69396ab9c0f588c71
  • Pointer size: 131 Bytes
  • Size of remote file: 111 kB