File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -746,7 +746,7 @@ class FusedLayerNorm(torch.nn.Module):
746
746
eps: a value added to the denominator for numerical stability. Default: 1e-5
747
747
elementwise_affine: a boolean value that when set to ``True``, this module
748
748
has learnable per-element affine parameters initialized to ones (for weights)
749
- and zeros (for biases). Default: `` True` `.
749
+ and zeros (for biases). Default: :obj:` True`.
750
750
751
751
Shape:
752
752
- Input: :math:`(N, *)`
@@ -844,8 +844,7 @@ class FusedRMSNorm(torch.nn.Module):
844
844
normalize over the last dimension which is expected to be of that specific size.
845
845
eps: a value added to the denominator for numerical stability. Default: 1e-5
846
846
elementwise_affine: a boolean value that when set to ``True``, this module
847
- has learnable per-element affine parameters initialized to ones (for weights)
848
- and zeros (for biases). Default: ``True``.
847
+ has learnable per-element affine parameters initialized to ones for weights. Default: :obj:`True`.
849
848
850
849
Shape:
851
850
- Input: :math:`(N, *)`
You can’t perform that action at this time.
0 commit comments