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
cb9552e2
Commit
cb9552e2
authored
Aug 02, 2017
by
Marianne Linhares Monteiro
Committed by
GitHub
Aug 02, 2017
Browse files
Fixing style
parent
1e41d029
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tutorials/image/cifar10_estimator/generate_cifar10_tfrecords.py
...als/image/cifar10_estimator/generate_cifar10_tfrecords.py
+2
-2
No files found.
tutorials/image/cifar10_estimator/generate_cifar10_tfrecords.py
View file @
cb9552e2
...
...
@@ -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.
...
...
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