Browse Source

!10750 fix the bug of op Conv2dBnFoldQuantOneConv

From: @lihongkang1
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @wuxuejian
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
9e2c30846e
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/nn/layer/quant.py

+ 3
- 0
mindspore/nn/layer/quant.py View File

@@ -348,6 +348,9 @@ class Conv2dBnFoldQuantOneConv(Cell):
padding (int): Implicit paddings on both sides of the input. Default: 0.
eps (float): Parameters for BatchNormal. Default: 1e-5.
momentum (float): Parameters for BatchNormal op. Default: 0.997.
dilation (int): Specifies the dilation rate to use for dilated convolution. Default: 1.
group (int): Splits filter into groups, `in_ channels` and `out_channels` must be
divisible by the number of groups. Default: 1.
has_bias (bool): Specifies whether the layer uses a bias vector. Default: False.
weight_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the
convolution kernel. Default: 'normal'.


Loading…
Cancel
Save