File tree 2 files changed +3
-3
lines changed
tutorials/image/cifar10_estimator
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ Before trying to run the model we highly encourage you to read all the README.
13
13
14
14
## Prerequisite
15
15
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
17
17
later.
18
18
19
19
2 . Download the CIFAR-10 dataset and generate TFRecord files using the provided
Original file line number Diff line number Diff line change @@ -71,11 +71,11 @@ def make_batch(self, batch_size):
71
71
"""Read the images and labels from 'filenames'."""
72
72
filenames = self .get_filenames ()
73
73
# Repeat infinitely.
74
- dataset = tf .contrib . data .TFRecordDataset (filenames ).repeat ()
74
+ dataset = tf .data .TFRecordDataset (filenames ).repeat ()
75
75
76
76
# Parse records.
77
77
dataset = dataset .map (
78
- self .parser , num_threads = batch_size , output_buffer_size = 2 * batch_size )
78
+ self .parser , num_parallel_calls = batch_size )
79
79
80
80
# Potentially shuffle records.
81
81
if self .subset == 'train' :
You can’t perform that action at this time.
0 commit comments