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

remove drop remainder from resnet pipeline. (#4416)

parent d41ed934
...@@ -80,7 +80,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer, ...@@ -80,7 +80,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer,
lambda value: parse_record_fn(value, is_training), lambda value: parse_record_fn(value, is_training),
batch_size=batch_size, batch_size=batch_size,
num_parallel_batches=1, num_parallel_batches=1,
drop_remainder=True)) 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
# will happen synchronously during run time. We prefetch here again to # will happen synchronously during run time. We prefetch here again to
......
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