Skip to content

Commit

Permalink
Fix (ptq): fix for act quantizers
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe5 committed Jul 17, 2023
1 parent ae7da18 commit 4c7afd9
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {
Expand Down

0 comments on commit 4c7afd9

Please # to comment.