Maggio33 commited on
Commit
f276d48
·
verified ·
1 Parent(s): c433d14

Model card: checkpoints listed so far (up to ckpt_17000.pt)

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