未验证 提交 ea6d6aab 编写于 作者: P pkulzc 提交者: GitHub

Merge pull request #4205 from vladpaunescu/master

Fixed Bug regarding tfrecord shuffling in object_detection
......@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册