Browse Source

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

GitOrigin-RevId: 747ae1646a
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
ae506ecfce
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      python_module/megengine/module/batchnorm.py

+ 1
- 1
python_module/megengine/module/batchnorm.py View File

@@ -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.



Loading…
Cancel
Save