Commit 5ac8b4dd authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

replace tf.contrib.data with tf.data

parent 44e40482
Pipeline #20415 canceled with stage
......@@ -102,7 +102,7 @@ def create_dataset_from_records(tfrecord_filenames,
if feature is None:
feature = DEFAULT_FEATURE
dataset = tf.contrib.data.TFRecordDataset(tfrecord_filenames)
dataset = tf.data.TFRecordDataset(tfrecord_filenames)
parser = partial(
example_parser,
feature=feature,
......@@ -144,7 +144,7 @@ def create_dataset_from_records_with_augmentation(
if feature is None:
feature = DEFAULT_FEATURE
dataset = tf.contrib.data.TFRecordDataset(tfrecord_filenames)
dataset = tf.data.TFRecordDataset(tfrecord_filenames)
parser = partial(
image_augmentation_parser,
feature=feature,
......
......@@ -67,7 +67,7 @@ def input_fn(mode, batch_size=1):
'convert_to_records.py first to convert the MNIST data to '
'TFRecord file format.')
dataset = tf.contrib.data.TFRecordDataset(tfrecords_files)
dataset = tf.data.TFRecordDataset(tfrecords_files)
# For training, repeat the dataset forever
if mode == tf.estimator.ModeKeys.TRAIN:
......
Supports Markdown
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