keras - 如何在 Keras 中为无限数据集设置批量大小和纪元值？

Question

我想将图像提供给 Keras CNN。该程序随机提供从网络下载的图像或随机像素值的图像。如何设置批量大小和纪元数？我的训练数据基本上是无限的。

Answer 1

即使您的数据集是无限的，您也必须同时设置批量大小和轮数。

对于批量大小，您可以通过反复试验使用适合您的 GPU/CPU RAM 的最大批量大小。例如，您可以尝试两个批量大小的幂，例如 32、64、128、256。

对于纪元数，这是一个始终必须针对特定问题进行调整的参数。您可以使用验证集进行训练，直到验证损失最大化，或者训练损失几乎恒定(收敛)。确保使用数据集的不同部分来决定何时停止训练。然后您可以报告另一个不同集(测试集)的最终指标。