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