Commit cb9552e2 authored by Marianne Linhares Monteiro's avatar Marianne Linhares Monteiro Committed by GitHub
Browse files

Fixing style

parent 1e41d029
......@@ -46,8 +46,7 @@ def _bytes_feature(value):
def _get_file_names():
"""Returns the file names expected to exist for training, validation
and evaluation in the input_dir."""
"""Returns the file names expected to exist in the input_dir."""
file_names = {}
file_names['train'] = ['data_batch_%d' % i for i in xrange(1, 5)]
file_names['validation'] = ['data_batch_5']
......@@ -81,6 +80,7 @@ def convert_to_tfrecord(input_files, output_file):
record_writer.write(example.SerializeToString())
record_writer.close()
def main(argv):
del argv # Unused.
......
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