experimental_allow_stateful is not necessary with make dataset fn

PiperOrigin-RevId: 274015143

experimental_allow_stateful is not necessary with make dataset fn
PiperOrigin-RevId: 274015143
399b778f · Hongkun Yu · A. Unique TensorFlower · bcb231f0 · 399b778f
Commit 399b778f authored Oct 10, 2019 by Hongkun Yu Committed by A. Unique TensorFlower Oct 10, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

official/vision/image_classification/imagenet_preprocessing.py ...ial/vision/image_classification/imagenet_preprocessing.py +1 -0

No files found.
--- a/official/vision/image_classification/imagenet_preprocessing.py
+++ b/official/vision/image_classification/imagenet_preprocessing.py
@@ -135,6 +135,7 @@ def process_record_dataset(dataset,

  options = tf.data.Options()
  options.experimental_slack = tf_data_experimental_slack
+  options.experimental_allow_stateful = True
  dataset = dataset.with_options(options)

  return dataset