Skip to content

GitLab

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

Remove repeating prefetch. 

PiperOrigin-RevId: 337242735
parent 44fc54a1
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
official/core/input_reader.py
View file @ bdf8299a
...@@ -277,8 +277,6 @@ class InputReader: ...@@ -277,8 +277,6 @@ class InputReader:
service=self._tf_data_service_address, service=self._tf_data_service_address,
job_name=self._tf_data_service_job_name)) job_name=self._tf_data_service_job_name))
dataset = dataset.prefetch(buffer_size=tf.data.experimental.AUTOTUNE)
if self._deterministic is not None: if self._deterministic is not None:
options = tf.data.Options() options = tf.data.Options()
options.experimental_deterministic = self._deterministic options.experimental_deterministic = self._deterministic
......
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