forked from mindspore-Ecosystem/mindspore
!9383 del freeze bn on conv2dbnwithfoldoneconv
From: @xiaoyisd Reviewed-by: @liangchenghui,@jjfeing Signed-off-by: @jjfeing
This commit is contained in:
commit
137e142b16
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue