Skip to content

GitLab

Commit 399b778f authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

experimental_allow_stateful is not necessary with make dataset fn 

PiperOrigin-RevId: 274015143
parent bcb231f0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/image_classification/imagenet_preprocessing.py
View file @ 399b778f
...@@ -135,6 +135,7 @@ def process_record_dataset(dataset, ...@@ -135,6 +135,7 @@ def process_record_dataset(dataset,
options = tf.data.Options() options = tf.data.Options()
options.experimental_slack = tf_data_experimental_slack options.experimental_slack = tf_data_experimental_slack
options.experimental_allow_stateful = True
dataset = dataset.with_options(options) dataset = dataset.with_options(options)
return dataset return dataset
......
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