Clear devices

parent a7ae458f
...@@ -328,7 +328,7 @@ class Trainer(object): ...@@ -328,7 +328,7 @@ class Trainer(object):
""" """
saver = self.architecture.load(self.model_from_file, clear_devices=True) saver = self.architecture.load(self.model_from_file, clear_devices=False)
# Loading training graph # Loading training graph
self.training_graph = tf.get_collection("training_graph")[0] self.training_graph = tf.get_collection("training_graph")[0]
......
Markdown is supported
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