Commit 2895148f authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

This might have solved the dask issue

parent b2044c7f
Pipeline #39985 failed with stage
in 13 minutes and 22 seconds
......@@ -107,8 +107,14 @@ class FaceNet(object):
return img[None, ...]
def load_model(self):
session_conf = tf.ConfigProto(
self.graph = tf.Graph()
self.session = tf.compat.v1.Session(graph=self.graph)
self.session = tf.compat.v1.Session(graph=self.graph, config=session_conf)
if self.model_path is None:
self.model_path = self.get_modelpath()
......@@ -189,7 +195,6 @@ class FaceNet(object):
def __setstate__(self, d):
# Handling unpicklable objects
self.__dict__ = d
def __getstate__(self):
# Handling unpicklable objects
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment