Maggio33 commited on
Commit
682f98f
·
verified ·
1 Parent(s): da7faba

Model card (run in progress)

Browse files
Files changed (1) hide show
  1. README.md +46 -0
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.