Unverified Commit ea6d6aab authored by pkulzc's avatar pkulzc Committed by GitHub
Browse files

Merge pull request #4205 from vladpaunescu/master

Fixed Bug regarding tfrecord shuffling in object_detection
parents 9f58547c e04280eb
......@@ -134,7 +134,7 @@ def read_dataset(file_read_func, decode_func, input_files, config):
file_read_func, cycle_length=config.num_readers,
block_length=config.read_block_length, sloppy=config.shuffle))
if config.shuffle:
records_dataset.shuffle(config.shuffle_buffer_size)
records_dataset = records_dataset.shuffle(config.shuffle_buffer_size)
tensor_dataset = records_dataset.map(
decode_func, num_parallel_calls=config.num_parallel_map_calls)
return tensor_dataset.prefetch(config.prefetch_size)
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