fixes tensorflow#2236: non-existant flag was used to point to the dat…

…aset (tensorflow#2238)
fastlater · Aug 18, 2017 · 527609e · 527609e
1 parent 02007a6
commit 527609e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ptn/eval_ptn.py b/ptn/eval_ptn.py
@@ -97,7 +97,7 @@ def main(argv=()):
     ## data ##
     ##########
     eval_data = model.get_inputs(
-        FLAGS.data_sst_path,
+        FLAGS.inp_dir,
         FLAGS.dataset_name,
         eval_params.eval_set,
         eval_params.batch_size,

diff --git a/ptn/eval_rotator.py b/ptn/eval_rotator.py
@@ -91,7 +91,7 @@ def main(argv=()):
     ## data ##
     ##########
     val_data = model.get_inputs(
-        FLAGS.data_sst_path,
+        FLAGS.inp_dir,
         FLAGS.dataset_name,
         'val',
         FLAGS.batch_size,