From 0e4a03363597bea7be0d4dbf5660d8c60b554901 Mon Sep 17 00:00:00 2001 From: TheCrazyT Date: Sun, 28 Jan 2024 08:41:28 +0100 Subject: [PATCH] Update documentation in quantize.py --- .../python/core/quantization/keras/quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_model_optimization/python/core/quantization/keras/quantize.py b/tensorflow_model_optimization/python/core/quantization/keras/quantize.py index 1d0c4154a..56f83f4ff 100644 --- a/tensorflow_model_optimization/python/core/quantization/keras/quantize.py +++ b/tensorflow_model_optimization/python/core/quantization/keras/quantize.py @@ -50,7 +50,7 @@ def quantize_scope(*args): # If your quantized model uses custom objects such as a specific `Quantizer`, # you can pass them to quantize_scope to deserialize your model. - with quantize_scope({'FixedRangeQuantizer', FixedRangeQuantizer} + with quantize_scope({'FixedRangeQuantizer': FixedRangeQuantizer}): loaded_model = tf.keras.models.load_model(keras_file) ```