Browse Source

docs(imperative): adjust bn doc

GitOrigin-RevId: 62f2b0338a
master
Megvii Engine Team 2 years ago
parent
commit
727be7eae5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/megengine/module/batchnorm.py

+ 1
- 1
imperative/python/megengine/module/batchnorm.py View File

@@ -290,7 +290,7 @@ class BatchNorm2d(_BatchNorm):
>>> oup = m(inp)
>>> print(m.weight.numpy().flatten(), m.bias.numpy().flatten())
[1. 1. 1. 1.] [0. 0. 0. 0.]
>>> # Without L`e`arnable Parameters
>>> # Without Learnable Parameters
>>> m = M.BatchNorm2d(4, affine=False)
>>> oup = m(inp)
>>> print(m.weight, m.bias)


Loading…
Cancel
Save