Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
e1364ca0
Commit
e1364ca0
authored
Sep 30, 2016
by
Christopher Shallue
Browse files
Close the TFRecordWriter after use.
parent
d7a4970e
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
0 deletions
+3
-0
im2txt/im2txt/data/build_mscoco_data.py
im2txt/im2txt/data/build_mscoco_data.py
+1
-0
inception/inception/data/build_image_data.py
inception/inception/data/build_image_data.py
+1
-0
inception/inception/data/build_imagenet_data.py
inception/inception/data/build_imagenet_data.py
+1
-0
No files found.
im2txt/im2txt/data/build_mscoco_data.py
View file @
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
()
...
...
inception/inception/data/build_image_data.py
View file @
e1364ca0
...
...
@@ -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
()
...
...
inception/inception/data/build_imagenet_data.py
View file @
e1364ca0
...
...
@@ -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
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment