Pick the most likely cause when training loss drops for 1k steps, plateaus for 2k, then drops again at step 3k
Answer A: a cosine schedule with restarts (or a manual LR bump) jumped the LR back up at step 3k, letting the optimizer escape the plateau; verify by plotting LR alongside loss.
Imagine driving up a winding mountain pass. Early on you make great progress, then you hit a long flat stretch where the engine seems to have given up. Suddenly at the next marker you start climbing again, smoothly. The road has not changed; what changed is that you shifted into a higher gear. In a training run, the step-size dial the trainer uses is the gear. Some schedules shift gears at planned points, and one of the most common is a cosine restart that decays toward a small floor then jumps back up. When the jump happens, the optimiser has fresh step size and can resume climbing toward a better minimum. The clean shape of the curve is the giveaway: smooth drop, long flat, smooth drop again at a specific step.
Detailed answer & concept explanation~8 min readEverything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example. Click to expand.
Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example. Click to expand.
Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example.
Everything important, quickly.
5 min: pick A, explain why the smooth drop, plateau, drop shape is the LR-restart fingerprint, walk through why each distractor fails its physical mechanism check, and prescribe the LR-over-loss overlay as the one-chart diagnostic.
Real products, models, and research that use this idea.
- PyTorch's CosineAnnealingWarmRestarts and Hugging Face Trainer's cosine_with_restarts produce exactly this curve shape and are common in long fine-tuning runs.
- DeepSpeed's training configuration exposes warm-restart schedules whose visual fingerprint matches this question's loss curve precisely.
- Weights & Biases and TensorBoard both default to plotting learning rate alongside loss for this diagnostic; serious training teams enable both panels by default.
- Modern pretraining runs for models like Llama 4 and Qwen 3.5 use cosine schedules and document the LR trace alongside loss in their training reports.
What an interviewer would ask next. Try answering before peeking at the approach.
QHow would you decide between a fixed cosine schedule and a cosine with restarts for a multi-day fine-tune?
QWhat other curve fingerprints should you recognise during loss-curve diagnostics?
Don't say thisRed flags and common mistakes that signal junior thinking. Click to expand.
Red flags and common mistakes that signal junior thinking. Click to expand.
Diagnosing this curve as 'the model finally learned the concept'. Loss curves do not have aha moments; clean drops at specific steps always trace back to a specific change in the optimization.
The night-before-the-interview bullets. Scan these on the way to the call.
Primary sources. Skim if you want the original framing.
Same topic, related formats. Practice these next.