"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "bbe3df01844b2e7eab1f9fe0d67fa3572a192b3d"
Unverified Commit 31587277 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Merge pull request #4748 from ricoms/master

update module to tensorflow 1.9.0
parents b8bfb196 c7bd6cd2
...@@ -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