Skip to content

S
seminar-breakout
Commit 8f4183e7 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

use tf.gfile in saver (fix #345)

parent 3917cfd5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/callbacks/saver.py
View file @ 8f4183e7
...@@ -45,7 +45,7 @@ class ModelSaver(Callback): ...@@ -45,7 +45,7 @@ class ModelSaver(Callback):
self.var_collections = var_collections self.var_collections = var_collections
if checkpoint_dir is None: if checkpoint_dir is None:
checkpoint_dir = logger.LOG_DIR checkpoint_dir = logger.LOG_DIR
assert os.path.isdir(checkpoint_dir), checkpoint_dir assert tf.gfile.IsDirectory(checkpoint_dir), checkpoint_dir
self.checkpoint_dir = checkpoint_dir self.checkpoint_dir = checkpoint_dir
def _setup_graph(self): def _setup_graph(self):
......
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