Browse Source

fix groupnorm bug and change globalbn parameter name

tags/v0.3.0-alpha
zhaojichen 6 years ago
parent
commit
f4de374124
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/normalization.py

+ 1
- 1
mindspore/nn/layer/normalization.py View File

@@ -350,7 +350,7 @@ class GlobalBatchNorm(_BatchNorm):
Tensor, the normalized, scaled, offset tensor, of shape :math:`(N, C_{out}, H_{out}, W_{out})`.

Examples:
>>> global_bn_op = nn.GlobalBatchNorm(num_features=3, group=4)
>>> global_bn_op = nn.GlobalBatchNorm(num_features=3, device_num_each_group=4)
>>> input = Tensor(np.random.randint(0, 255, [1, 3, 224, 224]), mindspore.float32)
>>> global_bn_op(input)
"""


Loading…
Cancel
Save