Browse Source

!9383 del freeze bn on conv2dbnwithfoldoneconv

From: @xiaoyisd
Reviewed-by: @liangchenghui,@jjfeing
Signed-off-by: @jjfeing
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
137e142b16
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      mindspore/nn/layer/quant.py

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

@@ -502,12 +502,12 @@ class Conv2dBnFoldQuantOneConv(Cell):
def extend_repr(self): def extend_repr(self):
s = 'in_channels={}, out_channels={}, kernel_size={}, stride={}, ' \ s = 'in_channels={}, out_channels={}, kernel_size={}, stride={}, ' \
'pad_mode={}, padding={}, dilation={}, group={}, ' \ 'pad_mode={}, padding={}, dilation={}, group={}, ' \
'fake={}, freeze_bn={}, momentum={}, quant_delay={}'.format(self.in_channels, self.out_channels,
self.kernel_size, self.stride,
self.pad_mode, self.padding, self.dilation,
self.group,
self.fake, self.freeze_bn, self.momentum,
self.fake_quant_weight.quant_delay)
'fake={}, momentum={}, quant_delay={}'.format(self.in_channels, self.out_channels,
self.kernel_size, self.stride,
self.pad_mode, self.padding, self.dilation,
self.group,
self.fake, self.momentum,
self.fake_quant_weight.quant_delay)
return s return s


def construct(self, x): def construct(self, x):


Loading…
Cancel
Save