From 5a26546b56ef959aa3102994fac2df1f95e273ff 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 +++++++----------- .../_op_impl/_custom_op/correction_mul_grad.py | 6 ------ 2 files changed, 7 insertions(+), 17 deletions(-) diff --git a/mindspore/nn/layer/quant.py b/mindspore/nn/layer/quant.py index 689f1c49556..fdbaa01ff6f 100644 --- a/mindspore/nn/layer/quant.py +++ b/mindspore/nn/layer/quant.py @@ -179,23 +179,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.ema: - 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={}({}, {}), ' \ diff --git a/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py b/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py index fc0ea2545e8..e79217f5213 100644 --- a/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +++ b/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py @@ -38,12 +38,6 @@ correction_mul_grad_op_info = TBERegOp("CorrectionMulGrad") \ .input(3, "running_std", None, "required", None) \ .output(0, "dx", True, "required", "all") \ .output(1, "d_batch_std", True, "required", "all") \ - .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, - DataType.F16_Default, DataType.F16_Default) \ - .dtype_format(DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, DataType.F16_5HD, - DataType.F16_5HD, DataType.F16_5HD) \ - .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, - DataType.F32_Default, DataType.F32_Default) \ .dtype_format(DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD, DataType.F32_5HD) \ .get_op_info()