|
|
@@ -136,6 +136,7 @@ class BatchNorm1d(_BatchNorm): |
|
|
eps (float): A value added to the denominator for numerical stability. Default: 1e-5. |
|
|
eps (float): A value added to the denominator for numerical stability. Default: 1e-5. |
|
|
momentum (float): A floating hyperparameter of the momentum for the |
|
|
momentum (float): A floating hyperparameter of the momentum for the |
|
|
running_mean and running_var computation. Default: 0.9. |
|
|
running_mean and running_var computation. Default: 0.9. |
|
|
|
|
|
affine (bool): A bool value when set to True, gamma and beta can be learnable. Default: True. |
|
|
gamma_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the gamma weight. |
|
|
gamma_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the gamma weight. |
|
|
The values of str refer to the function `initializer` including 'zeros', 'ones', 'xavier_uniform', |
|
|
The values of str refer to the function `initializer` including 'zeros', 'ones', 'xavier_uniform', |
|
|
'he_uniform', etc. Default: 'ones'. |
|
|
'he_uniform', etc. Default: 'ones'. |
|
|
@@ -187,6 +188,7 @@ class BatchNorm2d(_BatchNorm): |
|
|
eps (float): A value added to the denominator for numerical stability. Default: 1e-5. |
|
|
eps (float): A value added to the denominator for numerical stability. Default: 1e-5. |
|
|
momentum (float): A floating hyperparameter of the momentum for the |
|
|
momentum (float): A floating hyperparameter of the momentum for the |
|
|
running_mean and running_var computation. Default: 0.9. |
|
|
running_mean and running_var computation. Default: 0.9. |
|
|
|
|
|
affine (bool): A bool value when set to True, gamma and beta can be learnable. Default: True. |
|
|
gamma_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the gamma weight. |
|
|
gamma_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the gamma weight. |
|
|
The values of str refer to the function `initializer` including 'zeros', 'ones', 'xavier_uniform', |
|
|
The values of str refer to the function `initializer` including 'zeros', 'ones', 'xavier_uniform', |
|
|
'he_uniform', etc. Default: 'ones'. |
|
|
'he_uniform', etc. Default: 'ones'. |
|
|
|