Model card: checkpoints listed so far (up to ckpt_17000.pt)
Browse files
README.md
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- polish
|
| 6 |
+
- checkpoints
|
| 7 |
+
- training-history
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Slayer1B 1.07 — training checkpoints
|
| 11 |
+
|
| 12 |
+
Native training checkpoints of the run that is producing [Slayer1B 1.07](https://huggingface.co/SlayerLab/Slayer1B-1.07),
|
| 13 |
+
for resuming or studying training (full trainer state, not a Transformers/safetensors package): pretraining continued
|
| 14 |
+
from the final 1.06 checkpoint `ckpt_13848.pt` (sha256 `1fc7396c…`, published in
|
| 15 |
+
[Slayer1B-1.06-checkpoints](https://huggingface.co/SlayerLab/Slayer1B-1.06-checkpoints)). Each file holds the model
|
| 16 |
+
weights **and the optimizer state**.
|
| 17 |
+
|
| 18 |
+
**Status: training in progress.** The table below lists only the files that are in this repository now; further
|
| 19 |
+
checkpoints are added as training saves them (the planned end is step 22327, which will also be released as the final
|
| 20 |
+
1.07 weights in the model repository). Every file listed was compared after upload with the checksum list written by
|
| 21 |
+
the trainer: the LFS object id (sha256) and the size match.
|
| 22 |
+
|
| 23 |
+
| | |
|
| 24 |
+
|---|---|
|
| 25 |
+
| Start | `ckpt_13848.pt` (final state of 1.06, constant lr 6e-4) |
|
| 26 |
+
| Steps | 13849 → 22327 planned (8 479 steps × 1 048 576 tokens), 256 sequences × 4096 tokens per step |
|
| 27 |
+
| Schedule | **constant lr 6e-4**, no decay |
|
| 28 |
+
| Data | [SlayerLab/slayer-1.07-pretrain](https://huggingface.co/datasets/SlayerLab/slayer-1.07-pretrain), 8 891 005 707 tokens, one pass |
|
| 29 |
+
| Hardware | 4 × NVIDIA RTX 4090, DDP, gradient accumulation 32; global batch identical to the 8-GPU runs |
|
| 30 |
+
| Interruption | stopped on 4 November at step 17130 and resumed on 5 November from `ckpt_17000.pt`; steps 17001–17130 were trained twice, and the loss at step 17010 was identical in both runs. Checkpoints up to step 17000 were saved before the interruption (the resume started from `ckpt_17000.pt`); later checkpoints were saved after the resume |
|
| 31 |
+
| Trainer | `train_gpt_ref` r24.2 (sha256 `46f83b51…`); model definition `train_gpt_ref_r23.py` (unchanged) |
|
| 32 |
+
| Training curve and evaluations | https://track.fabryka.ai/run/4973c068-8dd5-59d8-b6c4-671ca3beb925 |
|
| 33 |
+
|
| 34 |
+
## SHA-256
|
| 35 |
+
|
| 36 |
+
| step | sha256 |
|
| 37 |
+
|---:|---|
|
| 38 |
+
| 17000 | `c4961c9ced633fef2efd8d28cc16217154f82b25f4a178ce31865b4523eed95b` |
|
| 39 |
+
|
| 40 |
+
Available training-time evaluations are in the
|
| 41 |
+
[Slayer1B 1.07 model card](https://huggingface.co/SlayerLab/Slayer1B-1.07).
|
| 42 |
+
|
| 43 |
+
## File format
|
| 44 |
+
|
| 45 |
+
The files are PyTorch pickles (`torch.save` dictionary of the trainer: `model`, `opt` (Muon + AdamW), `step`,
|
| 46 |
+
`config`); they contain only dictionaries, numbers, strings and tensors. Load with
|
| 47 |
+
`torch.load(path, map_location="cpu", weights_only=True)` (checked with torch 2.13 on `ckpt_17000.pt`; `mmap=True`
|
| 48 |
+
avoids reading the whole file into memory). Model class: `train_gpt_ref_r23.py` in the model repositories
|
| 49 |
+
(1.00–1.06). Every file is 8 860 287 151 bytes.
|
| 50 |
+
|
| 51 |
+
## License
|
| 52 |
+
|
| 53 |
+
Apache-2.0 for the checkpoints. The training data includes CC BY-SA sources (the share-alike parts of the Polish core
|
| 54 |
+
and legal texts, repeated from slayer-pl-8x3b); see the dataset cards of
|
| 55 |
+
[slayer-pl-8x3b](https://huggingface.co/datasets/SlayerLab/slayer-pl-8x3b) and
|
| 56 |
+
[slayer-1.07-pretrain](https://huggingface.co/datasets/SlayerLab/slayer-1.07-pretrain) for the licenses of the data.
|