Skip to content

GitLab

Commit 527609e0 authored by Oier Mees's avatar Oier Mees Committed by Reed
Browse files

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

parent 02007a65
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptn/eval_ptn.py
View file @ 527609e0
...@@ -97,7 +97,7 @@ def main(argv=()): ...@@ -97,7 +97,7 @@ def main(argv=()):
## data ## ## data ##
########## ##########
eval_data = model.get_inputs( eval_data = model.get_inputs(
FLAGS.data_sst_path, FLAGS.inp_dir,
FLAGS.dataset_name, FLAGS.dataset_name,
eval_params.eval_set, eval_params.eval_set,
eval_params.batch_size, eval_params.batch_size,
......
ptn/eval_rotator.py
View file @ 527609e0
...@@ -91,7 +91,7 @@ def main(argv=()): ...@@ -91,7 +91,7 @@ def main(argv=()):
## data ## ## data ##
########## ##########
val_data = model.get_inputs( val_data = model.get_inputs(
FLAGS.data_sst_path, FLAGS.inp_dir,
FLAGS.dataset_name, FLAGS.dataset_name,
'val', 'val',
FLAGS.batch_size, FLAGS.batch_size,
......
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