diff options
author | 2016-08-18 01:57:35 -0800 | |
---|---|---|
committer | 2016-08-18 03:05:01 -0700 | |
commit | 56ab88ce1bd59c312c34677eee242df45bda2336 (patch) | |
tree | 838a4ecba104546f978f8a726d14e0879a66adf8 /tensorflow/examples/how_tos | |
parent | 7b160cf8cd77b13176f8a734f438c60060e09b03 (diff) |
Parameterize validation size.
Change: 130618104
Diffstat (limited to 'tensorflow/examples/how_tos')
-rw-r--r-- | tensorflow/examples/how_tos/reading_data/convert_to_records.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/examples/how_tos/reading_data/convert_to_records.py b/tensorflow/examples/how_tos/reading_data/convert_to_records.py index 566d554e7f..f52e5bdddd 100644 --- a/tensorflow/examples/how_tos/reading_data/convert_to_records.py +++ b/tensorflow/examples/how_tos/reading_data/convert_to_records.py @@ -79,7 +79,8 @@ def main(argv): # Get the data. data_sets = mnist.read_data_sets(FLAGS.directory, dtype=tf.uint8, - reshape=False) + reshape=False, + validation_size=FLAGS.validation_size) # Convert to Examples and write the result to TFRecords. convert_to(data_sets.train, 'train') |