From 9dc985cb58ef7f1159d11daa3d96b0ad8206933d Mon Sep 17 00:00:00 2001 From: Adrian Frischknecht Date: Tue, 28 Nov 2023 10:07:56 +0100 Subject: [PATCH] Fixed POT quantization --- hannah/quantization/qconfig.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/hannah/quantization/qconfig.py b/hannah/quantization/qconfig.py index c0e53179..238429b7 100644 --- a/hannah/quantization/qconfig.py +++ b/hannah/quantization/qconfig.py @@ -207,9 +207,7 @@ def __init__( self.dtype = dtype if power_of_2: - self.quantization_function = PowerOf2Quantization( - bits, scale=scale, debug=self.debug - ) + self.quantization_function = PowerOf2Quantization(bits, debug=self.debug) else: self.quantization_function = SymmetricQuantization( bits, rounding_mode=rounding_mode, debug=self.debug