Small quantization doc corrections (apple#728)

RobertRiachi · Jun 19, 2020 · 0730319 · 0730319
1 parent 1afb9cb
commit 0730319
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)
 ```