I trained a 360M-param Python model from scratch on two workstation GPUs and wrote up every step, including the bugs
No transformers, no Trainer. Hand-written BPE tokenizer, Llama-style decoder, DDP training loop, SFT, all in plain PyTorch. 20B tokens on 2x RTX A4500, six days, about $21 of electricity. Base model gets 13.4% on HumanEval, the chat version 26.8% (Codex-12B was 28.8%). It is bad at everything that…
Read the full story at r/learnmachinelearning ↗
Timeline · 2 reports
- 2026-09-21 23:12 · r/deeplearning
I trained a 360M-param Python model from scratch on two workstation GPUs and wrote up every step, including the bugs - 2026-09-21 23:11 · r/learnmachinelearning
I trained a 360M-param Python model from scratch on two workstation GPUs and wrote up every step, including the bugs