tfmot.clustering.keras.strip_clustering
Strips clustering wrappers from the model.
tfmot.clustering.keras.strip_clustering(
model
)
Used in the notebooks
Once a model has been clustered, this method can be used
to restore the original model with the clustered weights.
Only sequential and functional models are supported for now.
Returns |
A keras model with clustering wrappers removed.
|
Raises |
ValueError
|
if the model is not a tf.keras.Model instance.
|
NotImplementedError
|
if the model is a subclass model.
|
Usage:
orig_model = tf.keras.Model(inputs, outputs)
clustered_model = cluster_weights(orig_model)
exported_model = strip_clustering(clustered_model)
The exported_model and the orig_model have the same structure.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2023-05-26 UTC.
[null,null,["Last updated 2023-05-26 UTC."],[],[]]