当训练数据对于fit（）太大时，如何知道何时在keras中使用fit_generator（）？

Question

在使用keras进行机器学习时，model.fit()用于训练数据较小的情况。当训练数据太大时，建议使用model.fit_generator()而不是model.fit()。如何知道数据大小何时过大？

Answer 1

尝试将训练数据存入内存时遇到内存错误，您必须切换到fit_generator()。快速生成数据（以及从磁盘读取数据）会产生额外的开销，因此在内存中的数据集上训练模型总是会更快。