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




Loading…
Cancel
Save