From 7b7fe538640c484388dbce4390d126d57c070f91 Mon Sep 17 00:00:00 2001 From: Elena Zhelezina Date: Thu, 24 Sep 2020 19:10:19 +0100 Subject: [PATCH] Small tidy up: Removed unused leftover variables. The variable layer.trainable_weights has all we need. Change-Id: I271319fd9ff09f71df3178bf2459bde55e09b72c --- .../python/core/clustering/keras/cluster_wrapper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py b/tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py index 7a0362e94..942096268 100644 --- a/tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py +++ b/tensorflow_model_optimization/python/core/clustering/keras/cluster_wrapper.py @@ -326,11 +326,11 @@ def trainable(self, value): @property def trainable_weights(self): - return self.layer.trainable_weights + self._trainable_weights + return self.layer.trainable_weights @property def non_trainable_weights(self): - return self.layer.non_trainable_weights + self._non_trainable_weights + return self.layer.non_trainable_weights @property def updates(self):