Skip to content

Unstable loss after a learning-rate increase

Question

After increasing the learning rate, training loss oscillates and occasionally becomes non-finite. What would you investigate first?

Short answer

Compare against the previous rate on identical data order and initialisation. Inspect gradient norms, input scales, loss implementation and mixed-precision behaviour, and check for non-finite inputs. A rate that overshoots can cause instability, but data or numerical errors can look similar. Reduce the rate, consider justified gradient clipping, and verify a small deterministic example before expanding the search.

Original question and answer written by SelfMadeQuant.com.

See the hint and the full worked solution in the question bank (free account).

Open the worked solution

Lessons behind this question

Start with a free lesson: Normal Distribution is part of a free topic. Create a free account, no card needed, to read it in full with worked solutions.

Read it free