Views
No views yet
| Original (HuggingFace) | This Run | |
|---|---|---|
| Tokenizer | GPT-2 (tiktoken), 10K vocab | GreedyPhrase, 65K vocab |
| Parameters | 4.3M | 15.0M |
| Hardware | 2 vCPU (CPU only) | RTX 2080 Ti (GPU) |
| Training time | 2 hours | ~2.2 hours |
| Tokens seen | 10.6M (2.3% of data) | 818M (3.3 epochs) |
| Best val loss | 2.0976 | 3.9352 |
| Throughput | 1,479 tok/s | 103,000 tok/s |
| Parameter | Value |
|---|---|
| Architecture | FlashLM v4 Bolt (ternary gated causal conv) |
| Hidden dim | 192 |
| Blocks | 6 |
| Conv kernel size | 8 |
| GLU expansion dim | 512 |
| Vocab size | 65,280 (padded from 65,218 actual) |
| Sequence length | 256 tokens |
| Effective batch size | 64 (micro=16, grad_accum=4) |
| Optimizer | AdamW (weight_decay=0.01) |
| Peak learning rate | 4e-3 |
| LR schedule | Cosine with 500-step warmup |
| Gradient clipping | 1.0 |
| Precision | AMP float16 |
| Total steps | 50,000 |
<|endoftext|> replaced with </s> (EOS token ID 3)Step Train Loss Val Loss
0 11.13 —
500 6.73 5.96
1000 5.46 5.12
2500 4.72 4.61
5000 4.43 4.39
10000 4.17 4.19
20000 4.03 4.03
30000 3.95 3.97
40000 3.92 3.95
50000 3.94 3.94
Best — 3.9352 (step 47500)| Metric | Value |
|---|---|
| Best validation loss | 3.9352 |
| Token-level perplexity | 51.17 |
| Bits per token | 5.68 |
| Bits per character (estimated) | 0.64 |
Once upon a time there was a little girl named Sarah. She was only three years old and loved exploring. One day Sarah went to the park with her mother. She saw a little boy playing with a ball.
Once upon a time there was a very deep lake. It was great! Every morning he would jump off the water and look for something wonderful.
Once upon a time there was a little girl named Mary. Mary loved animals, especially especially loved the ocean. Every day Mary would go out on a walk around the waves and swimming around on the beach.
The little dog wanted to protect his bone, so he held it up to the cat and tried to protect him. But the big cat was jealous. It wanted to take the bone from him, but it ran away.The cat was sad and began to cry. Then, he saw a big hole in the ground and started to shake it. The cat growled and tried to run away. The dog was scared and ran back to the cat. The cat saw the fox and was scared. The cat took the kitten and ran away. The dog was sad. The fox did not get the mitten anymore. The cat was happy and played with Spot and the other friends.
| File | Size | Description |
|---|---|---|
flashlm_v4_bolt_greedyphrase.pt | 58 MB | Final model (step 50,000) |
best.pt | 172 MB | Best checkpoint with optimizer state (step 47,500) |
checkpoint.pt | 172 MB | Latest periodic checkpoint |
tinystories.tokens | 496 MB | Tokenized dataset (uint16 binary) |
model.py | — | Model architecture |
train.py | — | Training script |