Pārlūkot izejas kodu

Merge pull request #885 from Henry-E/master

ensure output directory exists
Neal Wu 8 gadi atpakaļ
vecāks
revīzija
ff76fa6fce
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
  1. 3 0
      neural_programmer/neural_programmer.py

+ 3 - 0
neural_programmer/neural_programmer.py

@@ -195,6 +195,9 @@ def master(train_data, dev_data, utility):
     else:
       ckpt = tf.train.get_checkpoint_state(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))):
         print "create dir: ", model_dir
         tf.gfile.MkDir(model_dir)