diff --git a/tutorials/notebooks/example_keras_qat.ipynb b/tutorials/notebooks/example_keras_qat.ipynb index 3afd5f972..a25c8255a 100644 --- a/tutorials/notebooks/example_keras_qat.ipynb +++ b/tutorials/notebooks/example_keras_qat.ipynb @@ -328,4 +328,4 @@ }, "nbformat": 4, "nbformat_minor": 5 -} +} \ No newline at end of file diff --git a/tutorials/notebooks/example_keras_qat.py b/tutorials/notebooks/example_keras_qat.py index 17489fd76..1cbaa4fd7 100644 --- a/tutorials/notebooks/example_keras_qat.py +++ b/tutorials/notebooks/example_keras_qat.py @@ -55,7 +55,8 @@ def get_tpc(): quantization_preserving=False, fixed_scale=1.0, fixed_zero_point=0, - weights_multiplier_nbits=0) + weights_multiplier_nbits=0, + simd_size=None) default_configuration_options = tp.QuantizationConfigOptions([default_config]) tp_model = tp.TargetPlatformModel(default_configuration_options) diff --git a/tutorials/notebooks/example_pytorch_qat.py b/tutorials/notebooks/example_pytorch_qat.py index c74a1743d..01167bb14 100644 --- a/tutorials/notebooks/example_pytorch_qat.py +++ b/tutorials/notebooks/example_pytorch_qat.py @@ -173,7 +173,8 @@ def get_tpc(): quantization_preserving=False, fixed_scale=1.0, fixed_zero_point=0, - weights_multiplier_nbits=0) + weights_multiplier_nbits=0, + simd_size=None) default_configuration_options = tp.QuantizationConfigOptions([default_config]) tp_model = tp.TargetPlatformModel(default_configuration_options)