Adam Epsilon Underflow in FP16 Mixed Precision
When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights.
When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5).
- Symptom
RuntimeError: Function 'LogBackward' returned nan values in its ith output- Root cause
- When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights.
- Recommended fix
- Increase Adam Epsilon optimizer = torch.optim.AdamW(model.parameters(), lr=1e-4, eps=1e-5) Increasing epsilon to 1e-5 ensures it is safely representable in FP16 and avoids division by zero.
- How Denpex helps
- Denpex investigates Adam Epsilon Underflow in FP16 Mixed Precision using the evidence you provide or your connected workload collects. Earlier rank, host or application evidence is needed to distinguish an initiating failure from a downstream report.
What this failure is
Adam Epsilon Underflow in FP16 Mixed Precision is a Software failure seen during ML training runs. When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights. Common tags: Numerical Instability.
Is this what broke your run? Paste your log.
You're reading about Adam Epsilon Underflow in FP16 Mixed Precision. Paste your own traceback and relevant evidence for an investigation of your workload, with a next action or a specific missing fact. A reference entry does not establish your cause. No account or card for the free diagnosis. Review data handling before submitting sensitive logs.
Before uploading, review cloud data handling and local options.
Upload your log file. Large files are processed in your browser. Only selected evidence is submitted when you run diagnosis.
Why it happens (the mechanism)
Engineers often assume the learning rate is too high or gradient clipping is missing. However, gradient clipping only restricts the numerator; it does nothing to prevent division by zero in the denominator.
What you'll observe
- Loss is NaN
- RuntimeError: Function 'LogBackward' returned nan values in its ith output
- Attempted loss scale: 1, reducing to 1
Common symptoms and what they mean
| Symptom | Why it happens |
|---|---|
| Training proceeds normally for thousands of steps and then abruptly diverges. | When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights. |
| The loss suddenly becomes NaN during an optimizer step. | When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights. |
| Model weights become entirely NaN. | When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights. |
Which systems are affected
- PyTorch
- Transformers
- Mixed Precision Training
How to confirm this is the problem
Use this checklist to test the hypothesis against a small reproduction. No single line proves the root cause, so preserve the preceding events and compare one variable at a time.
- ✓Inspect the optimizer states to see if variance buffers are 0.
- ✓Enable PyTorch anomaly detection using torch.autograd.set_detect_anomaly(True).
- ✓Print the minimum and maximum values of the gradients right before the optimizer step.
Root cause
- When using FP16 for the optimizer state (or when the epsilon value itself is cast to FP16), the default Adam epsilon of 1e-8 underflows to 0 (since the smallest representable subnormal in FP16 is ~6e-5). When the moving average of squared gradients approaches zero, the denominator becomes exactly zero, causing a division by zero that yields NaN weights.
The fix and how to prevent it
Searchable error signature
RuntimeError: Function 'LogBackward' returned nan values in its ith outputUse this text as a lookup key in logs and upstream issue trackers. It is not presented as a captured customer log. Confirm the cause from your own preceding events, versions, configuration and the cited references.
The fix and the prevention pattern
The root cause is on this page and stays free. A free account adds the exact remediation steps, saved history, and the fix on every entry in the encyclopedia.
Sign up free. Unlock the full analysisNo credit card. Daily allowance follows verified trust tier. Instant access.
Diagnose this failure in VS Code
Select the traceback or open the failed terminal, then run Denpex locally to see the initiating rank, collateral failures, exact fix, and verification command without uploading the log.
Install the free VS Code extensionReferences
Don't just read the fix, diagnose your run
The encyclopedia tells you what went wrong. Denpex tells you what went wrong in YOUR training run. With your logs, your config, and your stack.