[Fix] sync zero_point bug

ModelTC · May 10, 2022 · fa8e075 · fa8e075
1 parent 0e875a8
commit fa8e075
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/mqbench/fake_quantize/adaround_quantizer.py b/mqbench/fake_quantize/adaround_quantizer.py
@@ -114,7 +114,7 @@ def forward(self, X):
                         self.ch_axis, self.quant_min, self.quant_max)
                 else:
                     X = torch.fake_quantize_per_tensor_affine(
-                        X, self.scale.item(), self.zero_point.item(),
+                        X, self.scale.item(), int(self.zero_point.item()),
                         self.quant_min, self.quant_max)
             else:
                 if not hasattr(self, 'alpha'):

diff --git a/mqbench/fake_quantize/dorefa.py b/mqbench/fake_quantize/dorefa.py
@@ -33,5 +33,5 @@ def forward(self, X):
                     self.ch_axis, self.quant_min, self.quant_max)
             else:
                 X = torch.fake_quantize_per_tensor_affine(
-                    X, self.scale.item(), self.zero_point.item(), self.quant_min, self.quant_max)
+                    X, self.scale.item(), int(self.zero_point.item()), self.quant_min, self.quant_max)
         return X
diff --git a/mqbench/fake_quantize/fixed.py b/mqbench/fake_quantize/fixed.py
@@ -34,7 +34,7 @@ def forward(self, X):
                     self.ch_axis, self.quant_min, self.quant_max)
             else:
                 X = torch.fake_quantize_per_tensor_affine(
-                    X, self.scale.item(), self.zero_point.item(),
+                    X, self.scale.item(), int(self.zero_point.item()),
                     self.quant_min, self.quant_max)
         return X
 

diff --git a/mqbench/fake_quantize/pact.py b/mqbench/fake_quantize/pact.py
@@ -47,6 +47,6 @@ def forward(self, X):
 
         if self.fake_quant_enabled[0] == 1:
             X = torch.fake_quantize_per_tensor_affine(
-                X, self.scale.item(), self.zero_point.item(), self.quant_min, self.quant_max)
+                X, self.scale.item(), int(self.zero_point.item()), self.quant_min, self.quant_max)
 
         return X