|
|
|
@ -178,23 +178,19 @@ class FakeQuantWithMinMax(Cell):
|
|
|
|
|
if self.is_ascend:
|
|
|
|
|
self.fake_quant = quant_fun(num_bits=self.num_bits,
|
|
|
|
|
symmetric=self.symmetric,
|
|
|
|
|
narrow_range=self.narrow_range,
|
|
|
|
|
training=self.training)
|
|
|
|
|
narrow_range=self.narrow_range)
|
|
|
|
|
else:
|
|
|
|
|
self.fake_quant = quant_fun(num_bits=self.num_bits,
|
|
|
|
|
ema=self.ema,
|
|
|
|
|
ema_decay=ema_decay,
|
|
|
|
|
quant_delay=quant_delay,
|
|
|
|
|
symmetric=self.symmetric,
|
|
|
|
|
narrow_range=self.narrow_range,
|
|
|
|
|
training=self.training)
|
|
|
|
|
if self.training:
|
|
|
|
|
self.ema_update = ema_fun(num_bits=self.num_bits,
|
|
|
|
|
ema=self.ema,
|
|
|
|
|
ema_decay=self.ema_decay,
|
|
|
|
|
symmetric=self.symmetric,
|
|
|
|
|
narrow_range=self.narrow_range,
|
|
|
|
|
training=self.training)
|
|
|
|
|
narrow_range=self.narrow_range)
|
|
|
|
|
self.ema_update = ema_fun(num_bits=self.num_bits,
|
|
|
|
|
ema=self.ema,
|
|
|
|
|
ema_decay=self.ema_decay,
|
|
|
|
|
symmetric=self.symmetric,
|
|
|
|
|
narrow_range=self.narrow_range)
|
|
|
|
|
|
|
|
|
|
def extend_repr(self):
|
|
|
|
|
s = 'num_bits={}, symmetric={}, narrow_range={}, ema={}({}), per_channel={}({}, {}), ' \
|
|
|
|
|