Changed the function to load checkpoints

parent 3287c426
......@@ -295,8 +295,9 @@ class Trainer(object):
file_name: Name of of the checkpoing
"""
self.saver = tf.train.import_meta_graph(file_name + ".meta", clear_devices=clear_devices)
self.saver.restore(self.session, file_name)
#self.saver = tf.train.import_meta_graph(file_name + ".meta", clear_devices=clear_devices)
self.saver = tf.train.import_meta_graph(file_name, clear_devices=clear_devices)
self.saver.restore(self.session, tf.train.latest_checkpoint(os.path.dirname(file_name)))
# Loading training graph
self.data_ph = tf.get_collection("data_ph")[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