From b7c1fbb4dc13ab4c52afc11761efa66c0f18d1ca Mon Sep 17 00:00:00 2001 From: baiyangfan Date: Fri, 11 Sep 2020 13:16:58 +0800 Subject: [PATCH] mode_init_bug --- mindspore/common/initializer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mindspore/common/initializer.py b/mindspore/common/initializer.py index d46fa141181..1da57645105 100644 --- a/mindspore/common/initializer.py +++ b/mindspore/common/initializer.py @@ -268,10 +268,10 @@ class XavierUniform(Initializer): Initialize the array with xavier uniform algorithm, and from a uniform distribution collect samples within U[-boundary, boundary] The boundary is defined as : - math:`boundary = gain * \sqrt{\frac{6}{n_{in} + n_{out}}}`. + where :math:`boundary = gain * \sqrt{\frac{6}{n_{in} + n_{out}}}`. - math:`n_{in}` is the number of input units in the weight tensor. - math:`n_{out}` is the number of output units in the weight tensor. + where :math:`n_{in}` is the number of input units in the weight tensor. + where :math:`n_{out}` is the number of output units in the weight tensor. Args: gain (Array): The array to be assigned. Default: 1. @@ -298,9 +298,9 @@ class HeUniform(Initializer): Initialize the array with He kaiming uniform algorithm, and from a uniform distribution collect samples within U[-boundary, boundary] The boundary is defined as : - math:`boundary = \sqrt{\frac{6}{n_{in}}}` + where :math:`boundary = \sqrt{\frac{6}{n_{in}}}`. - math:`n_{in}` is the number of input units in the weight tensor. + where :math:`n_{in}` is the number of input units in the weight tensor. Args: arr (Array): The array to be assigned.