Commit c7bd6cd2 authored by ricoms's avatar ricoms
Browse files

update module to tensorflow 1.9.0

parent 2dc6b914
......@@ -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
......
......@@ -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':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment