Skip to content

GitLab

Unverified Commit 3a624c29 authored by Taylor Robie's avatar Taylor Robie Committed by GitHub
Browse files

properly assign dataset in the second prefetch call (#4358)

parent ef59b3c7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/resnet_run_loop.py
View file @ 3a624c29
...@@ -88,7 +88,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer, ...@@ -88,7 +88,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer,
# critical training path. Setting buffer_size to tf.contrib.data.AUTOTUNE # critical training path. Setting buffer_size to tf.contrib.data.AUTOTUNE
# allows DistributionStrategies to adjust how many batches to fetch based # allows DistributionStrategies to adjust how many batches to fetch based
# on how many devices are present. # on how many devices are present.
dataset.prefetch(buffer_size=tf.contrib.data.AUTOTUNE) dataset = dataset.prefetch(buffer_size=tf.contrib.data.AUTOTUNE)
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