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.

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

A keras model or layer with clustering variables removed.

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