Skip to content
Snippets Groups Projects
Commit 66a416bf authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Removed the clear devices

parent 1a517ecd
Branches
Tags 0.1.4
No related merge requests found
Hide whitespace changes
Inline Side-by-side
bob/learn/tensorflow/network/SequenceNetwork.py
+ 2
− 2
View file @ 66a416bf
...@@ -322,8 +322,8 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)): ...@@ -322,8 +322,8 @@ class SequenceNetwork(six.with_metaclass(abc.ABCMeta, object)):
session = Session.instance().session session = Session.instance().session
self.sequence_net = pickle.loads(open(path+"_sequence_net.pickle").read()) self.sequence_net = pickle.loads(open(path+"_sequence_net.pickle").read())
saver = tf.train.import_meta_graph(path + ".meta", clear_devices=clear_devices) #saver = tf.train.import_meta_graph(path + ".meta", clear_devices=clear_devices)
#saver = tf.train.import_meta_graph(path + ".meta") saver = tf.train.import_meta_graph(path + ".meta")
saver.restore(session, path) saver.restore(session, path)
self.inference_graph = tf.get_collection("inference_graph")[0] self.inference_graph = tf.get_collection("inference_graph")[0]
self.inference_placeholder = tf.get_collection("inference_placeholder")[0] self.inference_placeholder = tf.get_collection("inference_placeholder")[0]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment