Maggio33 commited on
Commit
393ef98
·
verified ·
1 Parent(s): 52614cf

Model card (run in preparation)

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.03 — training checkpoints
11
+
12
+ Full trainer checkpoints of the run that produces Slayer1B 1.03: continued training of
13
+ [Slayer1B 1.02](https://huggingface.co/SlayerLab/Slayer1B-1.02) from its final checkpoint
14
+ ([`ckpt_5720.pt`](https://huggingface.co/SlayerLab/Slayer1B-1.02-checkpoints), sha256 `6b2a9b11…`, constant learning
15
+ rate). Each file holds the model weights **and the optimizer state**.
16
+
17
+ | | |
18
+ |---|---|
19
+ | Start | 1.02 `ckpt_5720.pt` (step 5720, constant lr 6e-4) |
20
+ | Steps | 7150 in total × 1 048 576 tokens = 7 497 318 400 tokens (+1430 steps = +1 499 463 680 tokens over 1.02) |
21
+ | Schedule | **constant lr 6e-4 to the end, no final decay** — the decay is deferred to the end of the whole Slayer1B line (pre-training continues on the whole prepared corpus of about 24 billion tokens) |
22
+ | Data | continues the pack-03 stream of [SlayerLab/slayer-pl-8x3b](https://huggingface.co/datasets/SlayerLab/slayer-pl-8x3b) (revision `d50df04a`) used by 1.02 from step 5390: steps 5721–7150 |
23
+ | Hardware | 4 × NVIDIA RTX 4090 (24 GB), DDP; global batch 256 × 4096 tokens as in 1.00–1.02 (micro-batch 2, gradient accumulation 32) |
24
+ | Checkpoints | every 150 steps from 5850 and the final step 7150; all are constant-lr states |
25
+ | Trainer | `train_gpt_ref_r24.py` (as in 1.02); the model definition is unchanged |
26
+
27
+ The run is in progress; checkpoints are added as they are written. Model card with evaluation results:
28
+ SlayerLab/Slayer1B-1.03 (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.02 model
34
+ repositories.
35
+
36
+ ## SHA-256
37
+
38
+ Recorded off the training process at the moment each checkpoint is written, before upload; files are downloaded back
39
+ and checked against these sums. The list is extended as the run progresses.
40
+
41
+ ```
42
+ ```
43
+
44
+ ## License
45
+
46
+ Apache-2.0.