RMSNorm¶
- class torchtune.modules.RMSNorm(dim: int, eps: float = 1e-06)[source]¶
Implements Root Mean Square Normalization introduced in https://arxiv.org/abs/1910.07467.
Reference implementation (used for correctness verfication) can be found here: https://github.com/facebookresearch/llama/blob/main/llama/model.py
- Parameters: