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