Skip to content

Commit 864de3c

Browse files
committed
Add Anti Zero-Drift functionality for Sparsity-Aware clustering (experimental)
* Fixed the signature of the public cluster_weights method
1 parent 24a6ac0 commit 864de3c

File tree

1 file changed

+0
-1
lines changed
  • tensorflow_model_optimization/python/core/clustering/keras

1 file changed

+0
-1
lines changed

tensorflow_model_optimization/python/core/clustering/keras/cluster.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@ def cluster_scope():
5656
def cluster_weights(to_cluster,
5757
number_of_clusters,
5858
cluster_centroids_init,
59-
preserve_sparsity=False,
6059
**kwargs):
6160
"""Modify a keras layer or model to be clustered during training.
6261

0 commit comments

Comments
 (0)