From 185d80930b491ee6a2ea3545bb14a05f9023c5d4 Mon Sep 17 00:00:00 2001 From: chenzomi Date: Thu, 11 Jun 2020 16:40:44 +0800 Subject: [PATCH] fake quant debug --- mindspore/nn/layer/quant.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/mindspore/nn/layer/quant.py b/mindspore/nn/layer/quant.py index ae01cab882..b843986b0e 100644 --- a/mindspore/nn/layer/quant.py +++ b/mindspore/nn/layer/quant.py @@ -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={}({}, {}), ' \