Commit 2cbd3375 authored by Chris Shallue's avatar Chris Shallue Committed by GitHub
Browse files

Merge pull request #483 from cshallue/writer

Close the TFRecordWriter after use.
parents 3bdc3253 e1364ca0
......@@ -286,6 +286,7 @@ def _process_image_files(thread_index, ranges, name, images, decoder, vocab,
(datetime.now(), thread_index, counter, num_images_in_thread))
sys.stdout.flush()
writer.close()
print("%s [thread %d]: Wrote %d image-caption pairs to %s" %
(datetime.now(), thread_index, shard_counter, output_file))
sys.stdout.flush()
......
......@@ -273,6 +273,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames,
(datetime.now(), thread_index, counter, num_files_in_thread))
sys.stdout.flush()
writer.close()
print('%s [thread %d]: Wrote %d images to %s' %
(datetime.now(), thread_index, shard_counter, output_file))
sys.stdout.flush()
......
......@@ -399,6 +399,7 @@ def _process_image_files_batch(coder, thread_index, ranges, name, filenames,
(datetime.now(), thread_index, counter, num_files_in_thread))
sys.stdout.flush()
writer.close()
print('%s [thread %d]: Wrote %d images to %s' %
(datetime.now(), thread_index, shard_counter, output_file))
sys.stdout.flush()
......
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