chenzomi
5 years ago
1 changed files with 7 additions and 11 deletions
Unified View
Diff Options
-
+7 -11mindspore/nn/layer/quant.py
+ 7
- 11
mindspore/nn/layer/quant.py
View File
| @@ -178,23 +178,19 @@ class FakeQuantWithMinMax(Cell): | |||||
| if self.is_ascend: | if self.is_ascend: | ||||
| self.fake_quant = quant_fun(num_bits=self.num_bits, | self.fake_quant = quant_fun(num_bits=self.num_bits, | ||||
| symmetric=self.symmetric, | symmetric=self.symmetric, | ||||
| narrow_range=self.narrow_range, | |||||
| training=self.training) | |||||
| narrow_range=self.narrow_range) | |||||
| else: | else: | ||||
| self.fake_quant = quant_fun(num_bits=self.num_bits, | self.fake_quant = quant_fun(num_bits=self.num_bits, | ||||
| ema=self.ema, | ema=self.ema, | ||||
| ema_decay=ema_decay, | ema_decay=ema_decay, | ||||
| quant_delay=quant_delay, | quant_delay=quant_delay, | ||||
| symmetric=self.symmetric, | 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): | def extend_repr(self): | ||||
| s = 'num_bits={}, symmetric={}, narrow_range={}, ema={}({}), per_channel={}({}, {}), ' \ | s = 'num_bits={}, symmetric={}, narrow_range={}, ema={}({}), per_channel={}({}, {}), ' \ | ||||