Skip to content

Commit

Permalink
fix build
Browse files Browse the repository at this point in the history
  • Loading branch information
ppwwyyxx committed Jul 16, 2018
1 parent d158a07 commit b17a098
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tensorpack/callbacks/saver.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,7 @@ def _before_train(self):
self.saver.export_meta_graph(
os.path.join(self.checkpoint_dir,
'graph-{}.meta'.format(time)),
collection_list=self.graph.get_all_collection_keys(),
clear_extraneous_savers=True)
collection_list=self.graph.get_all_collection_keys())

def _trigger(self):
try:
Expand Down

0 comments on commit b17a098

Please sign in to comment.