From 8b13b520c4d84ab93362ed3317d5619787d3d26a Mon Sep 17 00:00:00 2001 From: xiaoyisd Date: Wed, 2 Dec 2020 17:45:58 +0800 Subject: [PATCH] del freeze bn --- mindspore/nn/layer/quant.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/mindspore/nn/layer/quant.py b/mindspore/nn/layer/quant.py index 2221daa556..b04d4208da 100644 --- a/mindspore/nn/layer/quant.py +++ b/mindspore/nn/layer/quant.py @@ -502,12 +502,12 @@ class Conv2dBnFoldQuantOneConv(Cell): def extend_repr(self): s = 'in_channels={}, out_channels={}, kernel_size={}, stride={}, ' \ '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 def construct(self, x):