提交 ae506ecf 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

docs(mge/module): fix momentum equation in batch norm doc

GitOrigin-RevId: 747ae1646ade7cd80f25c50e9b20c4f332cb6930
上级 1c4a64b2
......@@ -136,7 +136,7 @@ class BatchNorm2d(_BatchNorm):
This :attr:`momentum` argument is different from one used in optimizer
classes and the conventional notion of momentum. Mathematically, the
update rule for running statistics here is
:math:`\hat{x}_\text{new} = (1 - \text{momentum}) \times \hat{x} + \text{momentum} \times x_t`,
:math:`\hat{x}_\text{new} = \text{momentum} \times \hat{x} + (1 - \text{momentum}) \times x_t`,
where :math:`\hat{x}` is the estimated statistic and :math:`x_t` is the
new observed value.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册