Skip to content

Commit 3158727

Browse files
authored
Merge pull request #4748 from ricoms/master
update module to tensorflow 1.9.0
2 parents b8bfb19 + c7bd6cd commit 3158727

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tutorials/image/cifar10_estimator/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ Before trying to run the model we highly encourage you to read all the README.
1313

1414
## Prerequisite
1515

16-
1. [Install](https://www.tensorflow.org/install/) TensorFlow version 1.2.1 or
16+
1. [Install](https://www.tensorflow.org/install/) TensorFlow version 1.9.0 or
1717
later.
1818

1919
2. Download the CIFAR-10 dataset and generate TFRecord files using the provided

tutorials/image/cifar10_estimator/cifar10.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -71,11 +71,11 @@ def make_batch(self, batch_size):
7171
"""Read the images and labels from 'filenames'."""
7272
filenames = self.get_filenames()
7373
# Repeat infinitely.
74-
dataset = tf.contrib.data.TFRecordDataset(filenames).repeat()
74+
dataset = tf.data.TFRecordDataset(filenames).repeat()
7575

7676
# Parse records.
7777
dataset = dataset.map(
78-
self.parser, num_threads=batch_size, output_buffer_size=2 * batch_size)
78+
self.parser, num_parallel_calls=batch_size)
7979

8080
# Potentially shuffle records.
8181
if self.subset == 'train':

0 commit comments

Comments
 (0)