Skip to content

GitLab

Commit cc0ad1cb authored by Alexandre Passos's avatar Alexandre Passos Committed by Taylor Robie
Browse files

Avoid using tf.contrib.data as it's not tf2-safe (#5755)

parent 2066a146
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/recommendation/data_preprocessing.py
View file @ cc0ad1cb
...@@ -639,7 +639,7 @@ def make_input_fn( ...@@ -639,7 +639,7 @@ def make_input_fn(
.format(epoch_metadata["batch_size"], batch_size)) .format(epoch_metadata["batch_size"], batch_size))
record_files_ds = tf.data.Dataset.list_files(record_files, shuffle=False) record_files_ds = tf.data.Dataset.list_files(record_files, shuffle=False)
interleave = tf.contrib.data.parallel_interleave( interleave = tf.data.experimental.parallel_interleave(
tf.data.TFRecordDataset, tf.data.TFRecordDataset,
cycle_length=4, cycle_length=4,
block_length=100000, block_length=100000,
......
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