Skip to content

Commit 3d2f895

Browse files
author
Taylor Robie
authored
remove drop remainder from resnet pipeline. (#4416)
1 parent d41ed93 commit 3d2f895

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

official/resnet/resnet_run_loop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def process_record_dataset(dataset, is_training, batch_size, shuffle_buffer,
8080
lambda value: parse_record_fn(value, is_training),
8181
batch_size=batch_size,
8282
num_parallel_batches=1,
83-
drop_remainder=True))
83+
drop_remainder=False))
8484

8585
# Operations between the final prefetch and the get_next call to the iterator
8686
# will happen synchronously during run time. We prefetch here again to

0 commit comments

Comments
 (0)