diff options
Diffstat (limited to 'tensorflow/examples/how_tos/reading_data/convert_to_records.py')
| -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') |