From 0730319f2c2294532f578c84dca88a6a0d65ff92 Mon Sep 17 00:00:00 2001 From: Kory Watson Date: Fri, 19 Jun 2020 10:27:48 -0700 Subject: [PATCH] Small quantization doc corrections (#728) --- examples/APIExamples.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/APIExamples.md b/examples/APIExamples.md index c709396bc..ae71ac8e6 100644 --- a/examples/APIExamples.md +++ b/examples/APIExamples.md @@ -295,9 +295,10 @@ class MyLayerSelector(QuantizedLayerSelector): ret = super(MyLayerSelector, self).do_quantize(layer) if not ret or layer.name == 'dense_2': return True + return ret selector = MyLayerSelector() quantized_model = quantize_weights( - mlmodel, 8, quantization_mode='linear', selector=selector) + model, 8, quantization_mode='linear', selector=selector) ```