Skip to content
Snippets Groups Projects

Refactor: removing scaling nodes

Files
7
@@ -17,7 +17,7 @@ from aidge_core import Log, Level
NB_SAMPLES = 1000 # max : 1000
SAMPLE_SHAPE = (1, 1, 28, 28)
MODEL_NAME = 'MiniResNet.onnx' # 'ConvNet.onnx'
ACCURACIES = (95.4, 94.5) # (97.9, 97.7)
ACCURACIES = (95.4, 94.4) # (97.9, 97.7)
NB_BITS = 4
# --------------------------------------------------------------
@@ -91,12 +91,13 @@ class test_ptq(unittest.TestCase):
# quantize the model
aidge_quantization.quantize_network(
self.model,
NB_BITS,
tensors,
clipping_mode=aidge_quantization.Clipping.MSE,
apply_rounding=True,
no_quantization=False,
optimize_signs=True,
single_shift=False
)
Loading