Skip to content

Commit 93120cd

Browse files
committed
Added docs to CQAT example.
Change-Id: I5f3fc8baf58e04585f1886151f9de6e47e157353
1 parent 5eb575e commit 93120cd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_model_optimization/g3doc/guide/combine/cqat_example.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,7 @@
253253
"clustering_params = {\n",
254254
" 'number_of_clusters': 8,\n",
255255
" 'cluster_centroids_init': CentroidInitialization.KMEANS_PLUS_PLUS\n",
256+
" 'cluster_per_channel': True,\n",
256257
"}\n",
257258
"\n",
258259
"clustered_model = cluster_weights(model, **clustering_params)\n",
@@ -597,7 +598,7 @@
597598
"source": [
598599
"## Apply post-training quantization and compare to CQAT model\n",
599600
"\n",
600-
"Next, we use post-training quantization (no fine-tuning) on the clustered model and check its accuracy against the CQAT model. This demonstrates why you would need to use CQAT to improve the quantized model's accuracy.\n",
601+
"Next, we use post-training quantization (no fine-tuning) on the clustered model and check its accuracy against the CQAT model. This demonstrates why you would need to use CQAT to improve the quantized model's accuracy. The difference could be not very visible, because the MNIST model is quite small and overparametrized.\n",
601602
"\n",
602603
"First, define a generator for the callibration dataset from the first 1000 training images."
603604
]

0 commit comments

Comments
 (0)