diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py index 16629804e..7982059f8 100644 --- 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 = {