Merge pull request #2926 from niboshi/doc-batch-norm-attrs

Fix BatchNormalization docstring
chainer · Jul 2, 2017 · a9f0c9a · a9f0c9a
2 parents d6a04fd + 825431f
commit a9f0c9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chainer/links/normalization/batch_normalization.py b/chainer/links/normalization/batch_normalization.py
@@ -53,8 +53,8 @@ class BatchNormalization(link.Link):
     Attributes:
         gamma (~chainer.Variable): Scaling parameter.
         beta (~chainer.Variable): Shifting parameter.
-        avg_mean (~chainer.Variable): Population mean.
-        avg_var (~chainer.Variable): Population variance.
+        avg_mean (numpy.ndarray or cupy.ndarray): Population mean.
+        avg_var (numpy.ndarray or cupy.ndarray): Population variance.
         N (int): Count of batches given for fine-tuning.
         decay (float): Decay rate of moving average. It is used on training.
         eps (float): Epsilon value for numerical stability. This value is added