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. ...@@ -13,7 +13,7 @@ Before trying to run the model we highly encourage you to read all the README.
## Prerequisite ## 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. later.
2. Download the CIFAR-10 dataset and generate TFRecord files using the provided 2. Download the CIFAR-10 dataset and generate TFRecord files using the provided
......
...@@ -71,11 +71,11 @@ class Cifar10DataSet(object): ...@@ -71,11 +71,11 @@ class Cifar10DataSet(object):
"""Read the images and labels from 'filenames'.""" """Read the images and labels from 'filenames'."""
filenames = self.get_filenames() filenames = self.get_filenames()
# Repeat infinitely. # Repeat infinitely.
dataset = tf.contrib.data.TFRecordDataset(filenames).repeat() dataset = tf.data.TFRecordDataset(filenames).repeat()
# Parse records. # Parse records.
dataset = dataset.map( 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. # Potentially shuffle records.
if self.subset == 'train': 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