Fix selective quantization Colab

PiperOrigin-RevId: 707188563
google-ai-edge · Dec 17, 2024 · 9b2aec6 · 9b2aec6
1 parent 353abd9
commit 9b2aec6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/colabs/selective_quantization_isnet.ipynb b/colabs/selective_quantization_isnet.ipynb
@@ -429,7 +429,8 @@
         "comparison_result = quantizer.validate(\n",
         "    test_data={'serving_default': [{'args_0': test_image}]},\n",
         "    error_metrics='median_diff_ratio',\n",
-        "    use_reference_kernel=True,\n",
+        "    use_xnnpack=False,\n",
+        "    num_threads=1,\n",
         ").save('', 'dynamic')"
       ]
     },