Browse Source

fix momentum bug

tags/v1.1.0
xiaoyisd 5 years ago
parent
commit
490e86be27
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/quant.py

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

@@ -842,7 +842,7 @@ class Conv2dBnWithoutFoldQuant(Cell):
channel_axis=channel_axis, channel_axis=channel_axis,
num_channels=out_channels, num_channels=out_channels,
quant_dtype=quant_dtype) quant_dtype=quant_dtype)
self.batchnorm = BatchNorm2d(out_channels, eps=eps, momentum=1-momentum)
self.batchnorm = BatchNorm2d(out_channels, eps=eps, momentum=momentum)


def construct(self, x): def construct(self, x):
weight = self.fake_quant_weight(self.weight) weight = self.fake_quant_weight(self.weight)


Loading…
Cancel
Save