tf.data.Dataset.shuffle的buffer_size要填多少的依据是什么？

来源：4-5 tf.data读取csv文件并与tf.keras结合使用

慕尼黑0536602

2020-03-19

老师，你好：我们这个：tf.data.Dataset.shuffle的buffer_size要填多少的依据是什么？
比方说，我们这个例子中，为什么shuffle_buffer_size=10000？
比方说，如果input_data >10000就填10000 ？
图片描述

写回答

1回答

正十七

2020-03-21

不是因为input_data 10000所以填的10000。

关于这个值的设置，其实主要和batch_size有关系，batch_size越大，这个值就应该越大，从而使得每一个batch中的元素不是那么固定，一般来说，可以考虑是batch_size的20~100倍左右即可。

慕尼黑0536602

好的，谢谢老师

2020-03-21

共1条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1