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
31587277
Unverified
Commit
31587277
authored
Jul 11, 2018
by
Toby Boyd
Committed by
GitHub
Jul 11, 2018
Browse files
Merge pull request #4748 from ricoms/master
update module to tensorflow 1.9.0
parents
b8bfb196
c7bd6cd2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
tutorials/image/cifar10_estimator/README.md
tutorials/image/cifar10_estimator/README.md
+1
-1
tutorials/image/cifar10_estimator/cifar10.py
tutorials/image/cifar10_estimator/cifar10.py
+2
-2
No files found.
tutorials/image/cifar10_estimator/README.md
View file @
31587277
...
...
@@ -13,7 +13,7 @@ Before trying to run the model we highly encourage you to read all the README.
## Prerequisite
1.
[
Install
](
https://www.tensorflow.org/install/
)
TensorFlow version 1.
2.1
or
1.
[
Install
](
https://www.tensorflow.org/install/
)
TensorFlow version 1.
9.0
or
later.
2.
Download the CIFAR-10 dataset and generate TFRecord files using the provided
...
...
tutorials/image/cifar10_estimator/cifar10.py
View file @
31587277
...
...
@@ -71,11 +71,11 @@ class Cifar10DataSet(object):
"""Read the images and labels from 'filenames'."""
filenames
=
self
.
get_filenames
()
# Repeat infinitely.
dataset
=
tf
.
contrib
.
data
.
TFRecordDataset
(
filenames
).
repeat
()
dataset
=
tf
.
data
.
TFRecordDataset
(
filenames
).
repeat
()
# Parse records.
dataset
=
dataset
.
map
(
self
.
parser
,
num_
threads
=
batch_size
,
output_buffer_size
=
2
*
batch_size
)
self
.
parser
,
num_
parallel_calls
=
batch_size
)
# Potentially shuffle records.
if
self
.
subset
==
'train'
:
...
...
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