Model card (run in progress)
Browse files
README.md
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- polish
|
| 6 |
+
- checkpoints
|
| 7 |
+
- training-history
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Slayer1B 1.02 — training checkpoints
|
| 11 |
+
|
| 12 |
+
Full trainer checkpoints of the run that produces Slayer1B 1.02: continued training of
|
| 13 |
+
[Slayer1B 1.01](https://huggingface.co/SlayerLab/Slayer1B-1.01) from its last checkpoint before the learning-rate decay
|
| 14 |
+
([`ckpt_2250.pt`](https://huggingface.co/SlayerLab/Slayer1B-1.01-checkpoints), sha256 `aeddfb02…`), with a new total
|
| 15 |
+
budget of twice the tokens of 1.01. Each file holds the model weights **and the optimizer state**.
|
| 16 |
+
|
| 17 |
+
| | |
|
| 18 |
+
|---|---|
|
| 19 |
+
| Start | 1.01 `ckpt_2250.pt` (step 2250, constant lr 6e-4) |
|
| 20 |
+
| Steps | 5720 in total × 1 048 576 tokens = 5 997 854 720 tokens (≈ 5.6 tokens per parameter; 1.01: 2860 steps) |
|
| 21 |
+
| Schedule | constant lr 6e-4 through step 4577, then 1 − sqrt decay over steps 4578–5720 to lr / 10 |
|
| 22 |
+
| Data | steps 1–1801: pack-00 stream (as 1.00 and 1.01); 1802–3594: pack-01 stream (as 1.01, continued); 3595–5389: pack-02; 5390–5720: pack-03 — all from [SlayerLab/slayer-pl-8x3b](https://huggingface.co/datasets/SlayerLab/slayer-pl-8x3b) revision `d50df04a`; shared components of each pack disjoint from those already used |
|
| 23 |
+
| Checkpoints | every 150 steps from 2400, **step 4576** (end of the constant-lr phase, starting point for further continued training) and the final step 5720 |
|
| 24 |
+
| Trainer | `train_gpt_ref_r24.py` = the 1.00/1.01 trainer plus a chain of data streams and extra checkpoint steps; batches up to step 3594 are identical to those of 1.01's data order |
|
| 25 |
+
| Training curve and evaluations during training | https://track.fabryka.ai/run/21ae1234-82c2-42b7-887b-cbcc69901ae1 |
|
| 26 |
+
|
| 27 |
+
The run is in progress; checkpoints are added as they are written. Model card with evaluation results:
|
| 28 |
+
SlayerLab/Slayer1B-1.02 (after the run).
|
| 29 |
+
|
| 30 |
+
## File format
|
| 31 |
+
|
| 32 |
+
`torch.save` dictionary of the trainer: `model`, `opt` (Muon + AdamW), `step`, `config`. Load with
|
| 33 |
+
`torch.load(path, map_location="cpu", weights_only=True)`. Model class: `train_gpt_ref_r23.py` in the 1.00/1.01 model
|
| 34 |
+
repositories (the architecture is unchanged).
|
| 35 |
+
|
| 36 |
+
## SHA-256
|
| 37 |
+
|
| 38 |
+
Recorded off the training node at the moment each checkpoint is written, before upload; files are downloaded back and
|
| 39 |
+
checked against these sums. The list is extended as the run progresses.
|
| 40 |
+
|
| 41 |
+
```
|
| 42 |
+
```
|
| 43 |
+
|
| 44 |
+
## License
|
| 45 |
+
|
| 46 |
+
Apache-2.0.
|