fixes #2236: non-existant flag was used to point to the dataset (#2238)

527609e0 · Oier Mees · Reed · 02007a65 · 527609e0 · 527609e0
Commit 527609e0 authored Aug 18, 2017 by Oier Mees Committed by Reed Aug 17, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

ptn/eval_ptn.py ptn/eval_ptn.py +1 -1

ptn/eval_rotator.py ptn/eval_rotator.py +1 -1

No files found.
--- 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,

--- 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,