Commit 8f4183e7 authored by Yuxin Wu's avatar Yuxin Wu

use tf.gfile in saver (fix #345)

parent 3917cfd5
......@@ -45,7 +45,7 @@ class ModelSaver(Callback):
self.var_collections = var_collections
if checkpoint_dir is None:
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
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