Code · architecture · study mode
Batch size starts small (8×2048×8 tokens = 134M tokens/step) then grows in two steps up to 24×2048×8 (402M). Small…
Full explanation below the code →
Batch size starts small (8×2048×8 tokens = 134M tokens/step) then grows in two steps up to 24×2048×8 (402M). Small batches early → high gradient noise → more exploration. Large batches late → low noise → convergent fine-tuning.
Why does growing the batch size during training improve convergence compared to keeping it constant?