Skip to content

GitLab

Commit eff33131 authored by Toby Boyd's avatar Toby Boyd
Browse files

fix num_parallel_batches

parent 1f21b69e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/resnet_run_loop.py
View file @ eff33131
...@@ -91,7 +91,7 @@ def process_record_dataset(dataset, ...@@ -91,7 +91,7 @@ def process_record_dataset(dataset,
tf.contrib.data.map_and_batch( tf.contrib.data.map_and_batch(
lambda value: parse_record_fn(value, is_training, dtype), lambda value: parse_record_fn(value, is_training, dtype),
batch_size=batch_size, batch_size=batch_size,
num_parallel_calls=num_parallel_batches, num_parallel_batches=num_parallel_batches,
drop_remainder=False)) drop_remainder=False))
# Operations between the final prefetch and the get_next call to the iterator # Operations between the final prefetch and the get_next call to the iterator
......
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