diff --git a/bob/bio/face/embeddings/facenet_sanderberg.py b/bob/bio/face/embeddings/facenet_sanderberg.py index ceaf9a84c31a8a0b81dd16b63fc6b28617eef3eb..ada60509026b3017e4f9d3f17ee375ef08807981 100644 --- a/bob/bio/face/embeddings/facenet_sanderberg.py +++ b/bob/bio/face/embeddings/facenet_sanderberg.py @@ -223,7 +223,6 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator): def __setstate__(self, d): # Handling unpicklable objects self.__dict__ = d - self.loaded = False def __getstate__(self): import tensorflow as tf @@ -236,6 +235,7 @@ class FaceNetSanderberg(TransformerMixin, BaseEstimator): d.pop("images_placeholder") if "images_placeholder" in d else None d.pop("phase_train_placeholder") if "phase_train_placeholder" in d else None tf.compat.v1.reset_default_graph() + self.loaded = False return d def _more_tags(self): diff --git a/bob/bio/face/embeddings/tensorflow_compat_v1.py b/bob/bio/face/embeddings/tensorflow_compat_v1.py index 0936523f62e6b148afe7d78c21229b1c3c4f4527..5659cb3f1a7e1c1c95923804dde222e7401fa374 100644 --- a/bob/bio/face/embeddings/tensorflow_compat_v1.py +++ b/bob/bio/face/embeddings/tensorflow_compat_v1.py @@ -122,8 +122,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator): def __setstate__(self, d): # Handling unpicklable objects - self.__dict__ = d - self.loaded = False + self.__dict__ = d def __getstate__(self): import tensorflow as tf @@ -134,6 +133,7 @@ class TensorflowCompatV1(TransformerMixin, BaseEstimator): d.pop("input_tensor", None) d.pop("embedding", None) tf.compat.v1.reset_default_graph() + self.loaded = False return d # def __del__(self):