We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 45b15be commit 77d074cCopy full SHA for 77d074c
tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py
@@ -107,9 +107,9 @@ def __init__(self,
107
108
# Stores the pairs of weight names and their respective sparsity masks
109
self.sparsity_masks = {}
110
- self.zero_idx = {}
111
112
# Stores the pairs of weight names and the zero centroids
+ self.zero_idx = {}
113
114
# Map weight names to original clusterable weights variables
115
# Those weights will still be updated during backpropagation
0 commit comments