Skip to content

Commit

Permalink
[Fix] Remove symmetric_range from kwargs (#97)
Browse files Browse the repository at this point in the history
Co-authored-by: xuexingyuan <xuexingyuan@sensetime.com>
  • Loading branch information
estherxue and xuexingyuan committed May 18, 2022
1 parent 9483cbb commit a5cabb7
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion mqbench/scheme.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,16 @@ def __init__(self, symmetry=True, per_channel=False, pot_scale=False, bit=8, **k
self.torch_qscheme = torch.per_channel_symmetric if self.symmetry else torch.per_channel_affine
else:
self.torch_qscheme = torch.per_tensor_symmetric if self.symmetry else torch.per_tensor_affine
if 'symmetric_range' in kwargs:
self.symmetric_range = kwargs['symmetric_range']
del kwargs['symmetric_range']
else:
self.symmetric_range = False
self.kwargs = kwargs

def to_observer_params(self):
naive_para = {
'quant_min': (-2 ** (self.bit - 1) + 1 if self.kwargs.get('symmetric_range', False) else -2 ** (self.bit - 1)) \
'quant_min': (-2 ** (self.bit - 1) + 1 if self.symmetric_range else -2 ** (self.bit - 1)) \
if self.symmetry else 0,
'quant_max': 2 ** (self.bit - 1) - 1 if self.symmetry else 2 ** self.bit - 1,
'dtype': torch.qint8 if self.symmetry else torch.quint8,
Expand Down

0 comments on commit a5cabb7

Please sign in to comment.