Skip to content

Commit 607599d

Browse files
saberkuntensorflower-gardener
authored andcommitted
Use None for tf.data seed when tf.data service is enabled.
PiperOrigin-RevId: 358727239
1 parent 9ff7639 commit 607599d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

official/core/input_reader.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,10 @@ def __init__(self,
111111
self._parser_fn = parser_fn
112112
self._transform_and_batch_fn = transform_and_batch_fn
113113
self._postprocess_fn = postprocess_fn
114-
self._seed = _get_random_integer()
114+
# When tf.data service is enabled, each data service worker should get
115+
# different random seeds. Thus, we set `seed` to None.
116+
self._seed = (None
117+
if params.enable_tf_data_service else _get_random_integer())
115118

116119
self._enable_tf_data_service = (
117120
params.enable_tf_data_service and params.tf_data_service_address)

0 commit comments

Comments
 (0)