make checkpoint_dir if not exists. fix #3

carpedm20 · Mar 23, 2016 · 2ea5436 · 2ea5436
1 parent 561ccb0
commit 2ea5436
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -1,3 +1,4 @@
+import os
 import pprint
 import tensorflow as tf
 
@@ -30,6 +31,9 @@ def main(_):
     count = []
     word2idx = {}
 
+    if not os.path.exists(FLAGS.checkpoint_dir):
+      os.makedirs(FLAGS.checkpoint_dir)
+
     train_data = read_data('%s/%s.train.txt' % (FLAGS.data_dir, FLAGS.data_name), count, word2idx)
     valid_data = read_data('%s/%s.valid.txt' % (FLAGS.data_dir, FLAGS.data_name), count, word2idx)
     test_data = read_data('%s/%s.test.txt' % (FLAGS.data_dir, FLAGS.data_name), count, word2idx)