Denoiser
Output projection
×learned
Map the modulated features to the predicted clean latents.
- in
- [B batch, T positions, 768 d]
- out
- [B batch, T positions, 768 d]
- parameters
- 590,592
Parameters
590,592 parameters per instance.
| Component | Calculation | Parameters |
|---|---|---|
| Weight matrix | 589,824 | |
| Output bias | 768 | |
| Per-instance total | 590,592 | |
The reference module holds the weight output by input, [out, in].
Derived from the configured matrix dimensions; checkpoint tensors have not been independently counted.
Explanation
Its weights and bias start at zero, so an untrained denoiser predicts zero latents.
Sources1
Direct evidence
The output layer: gain-only LayerNorm, a shift and a scale from the time condition, and an affine projection initialized at zero.
The implementation, pinned at revision e18c42acdfb7, not yet public