Skip to content

GitLab

Commit ff76fa6f authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #885 from Henry-E/master 

ensure output directory exists
parents 99462f6d 6601008f
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
neural_programmer/neural_programmer.py
View file @ ff76fa6f
...@@ -195,6 +195,9 @@ def master(train_data, dev_data, utility): ...@@ -195,6 +195,9 @@ def master(train_data, dev_data, utility):
else: else:
ckpt = tf.train.get_checkpoint_state(model_dir) ckpt = tf.train.get_checkpoint_state(model_dir)
print "model dir: ", model_dir print "model dir: ", model_dir
if (not (tf.gfile.IsDirectory(utility.FLAGS.output_dir))):
print "create dir: ", utility.FLAGS.output_dir
tf.gfile.MkDir(utility.FLAGS.output_dir)
if (not (tf.gfile.IsDirectory(model_dir))): if (not (tf.gfile.IsDirectory(model_dir))):
print "create dir: ", model_dir print "create dir: ", model_dir
tf.gfile.MkDir(model_dir) tf.gfile.MkDir(model_dir)
......
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