tfmot.experimental.combine.strip_clustering_cqat

Strip clustering variables from the model.

tfmot.experimental.combine.strip_clustering_cqat(
    to_strip
)

During cluster-preserve quantization aware training (CQAT), centroids, cluster associations, and original weights are added to the training graph. After the CQAT is done, these variables should be removed and the layer with the clustered weights should be restored.

Arguments
to_strip	A tf.keras.Model instance with clustered layers or a tf.keras.layers.Layer instance

Returns
A keras model or layer with clustering variables removed.

Raises
ValueError	if the model is not a tf.keras.Model instance.
NotImplementedError	if the model is a subclassed model.