Wrong Weight Initialization
Wrong weight initialization causes training instability with NaN losses, slow convergence, or dead neurons.
Wrong weight initialization causes training instability with NaN losses, slow convergence, or dead neurons.
- Root cause
- Weights initialized too small causing vanishing gradients. Weights initialized too large causing exploding gradients. Activation function not suited to initialization scheme.
- Recommended fix
- Use PyTorch default initialization for the layer type
- How Denpex helps
- Denpex investigates Wrong Weight Initialization 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
Wrong Weight Initialization is a Training Stability failure seen during ML training runs. Wrong weight initialization causes training instability with NaN losses, slow convergence, or dead neurons. Common tags: Weight Initialization, Kaiming, Xavier, Training Stability.
Is this what broke your run? Paste your log.
You're reading about Wrong Weight Initialization. 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)
Weights initialized too small causing vanishing gradients. Weights initialized too large causing exploding gradients. Activation function not suited to initialization scheme. No weight initialization specified in model code. Taken together, these mechanisms explain why the failure is reproducible, why it tends to surface on specific workloads or scales, and why generic mitigation attempts often fall short without addressing the underlying cause.
What you'll observe
- Loss is NaN from the first step
- Loss decreases very slowly or not at all
- Many neurons output 0 (dead ReLU)
Common symptoms and what they mean
| Symptom | Why it happens |
|---|---|
| loss: nan in first step | Weights initialized too small causing vanishing gradients |
| loss stays constant for thousands of steps | Weights initialized too large causing exploding gradients |
| Model produces constant outputs for all inputs | Activation function not suited to initialization scheme |
Which systems are affected
- Training from scratch (not fine-tuning)
- Training with new model architecture
- Models with deep networks
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.
- ✓Reproduce the failure from a clean checkpoint/seed: the symptom must appear without warm-up state from a previous run.
- ✓Verified signal present: loss: nan in first step
- ✓Verified signal present: loss stays constant for thousands of steps
- ✓Verified signal present: Model produces constant outputs for all inputs
- ✓A targeted fix from the "How to fix it" section eliminates or substantially reduces the symptom within one validation pass.
Root cause
- Weights initialized too small causing vanishing gradients
- Weights initialized too large causing exploding gradients
- Activation function not suited to initialization scheme
- No weight initialization specified in model code
The fix and how to prevent it
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 extensionRelated failures to investigate next
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.