Fix (examples/imagenet/ptq): fix for unsigned act quantizers (#675)

Xilinx · Jul 18, 2023 · d2f3832 · d2f3832
1 parent b1c6839
commit d2f3832
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -275,12 +275,12 @@ def kwargs_prefix(prefix, weight_kwargs):
  torch.nn.ReLU: (qnn.QuantReLU, {
  **quant_act_kwargs, 'signed': False}),
  torch.nn.ReLU6: (qnn.QuantReLU, {
- **quant_act_kwargs, 'signed': False, 'max_val': 6.}),
+ **quant_act_kwargs, 'signed': False}),
  torch.nn.Sigmoid: (qnn.QuantSigmoid, {
  **quant_act_kwargs, 'signed': False}),}
  quant_identity_map = {
  'signed': (qnn.QuantIdentity, {
- **quant_act_kwargs, 'signed': True}),
+ **quant_act_kwargs}),
  'unsigned': (qnn.QuantIdentity, {
  **quant_act_kwargs, 'signed': False}),}
  quant_layerwise_layer_map = {