We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d41ed93 commit 3d2f895Copy full SHA for 3d2f895
official/resnet/resnet_run_loop.py
@@ -80,7 +80,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer,
80
lambda value: parse_record_fn(value, is_training),
81
batch_size=batch_size,
82
num_parallel_batches=1,
83
- drop_remainder=True))
+ drop_remainder=False))
84
85
# Operations between the final prefetch and the get_next call to the iterator
86
# will happen synchronously during run time. We prefetch here again to
0 commit comments