Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "be2722e19b437b78189f2e6db6c6ae181b575bf8"
Commit 55ec8194 authored by Allen Wang's avatar Allen Wang Committed by A. Unique TensorFlower
Browse files

Update tfds readconfig parameter 

PiperOrigin-RevId: 309455478
parent 38b92207
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/image_classification/dataset_factory.py
View file @ 55ec8194
......@@ -342,7 +342,7 @@ class DatasetBuilder:
decoders['image'] = tfds.decode.SkipDecoding()
read_config = tfds.ReadConfig(
interleave_parallel_reads=64,
interleave_cycle_length=64,
interleave_block_length=1)
dataset = builder.as_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