大型数据集的最佳批处理大小和优化程序是什么?

时间:2019-07-16 06:27:41

标签: python-3.x keras jupyter-notebook

我正在处理45000张图像的数据集。

1)这样的大型数据集应该使用多少批处理大小??

2)我应该使用哪种优化器,例如Adam或SGD,我总是有些困惑,总是选择优化器。

环境:I7-8750h和GTX 1060(6GB)

1 个答案:

答案 0 :(得分:3)

45K不是一个大数据集,它小于MNIST数据集(60K图像)。

但是无论如何,没有人可以提出批量大小和优化器建议,这是您必须在自己的数据集和模型中进行调整的地方。您可以从Adam开始,看看模型是否收敛,如果不收敛到SGD。