Commit 35c5a4e8 authored by Yuxin Wu's avatar Yuxin Wu

avoid saving metagraph too many times

parent 785e01e2
......@@ -29,6 +29,7 @@ class ModelSaver(Callback):
var_list=ModelSaver._get_vars(),
max_to_keep=self.keep_recent,
keep_checkpoint_every_n_hours=self.keep_freq)
self.meta_graph_written = False
@staticmethod
def _get_vars():
......@@ -51,7 +52,10 @@ class ModelSaver(Callback):
self.saver.save(
tf.get_default_session(),
self.path,
global_step=self.global_step)
global_step=self.global_step,
write_meta_graph=not self.meta_graph_written)
if not self.meta_graph_written:
self.meta_graph_written = True
class MinSaver(Callback):
def __init__(self, monitor_stat):
......
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