the.ai

Optimization / Schedules

verified

Warmup

At the very start of training the parameters are random and the gradients are large and meaningless. Taking full-size steps then can wreck the run in the first few hundred iterations. Warmup ramps the learning rate up from near zero over the first stretch, so the model gets its bearings before it starts sprinting.

Viz primitive · loss-curvesteps = 1000 · lr = 0.004 · warmup = 100
loss
step 0dashed = no warmup1000

Loss over 1000 training steps, starting near 6.5. It falls to about 1.77, with 95% of the total improvement arriving in the first half. A second line shows no warmup, ending at about 1.77.

Loss through the opening of a run, with and without a ramp. Drag the warmup length to watch the ramp stretch — the spike is gone the moment there is any warmup at all, and what the extra length buys is a longer gentle opening rather than a calmer one.

100

Reviewed by opendroid · 2026-08-04