当训练数据对于fit()来说太大时,如何知道何时在keras中使用fit_generator()?

use*_*789 3 machine-learning python-3.x keras

当使用keras进行机器学习时,model.fit()在训练数据很小时使用.当训练数据太大时,model.fit_generator()建议不要使用model.fit().如何知道数据大小何时变得过大?

sdc*_*cbr 5

当您尝试将训练数据带入内存时遇到内存错误的那一刻,您将不得不切换到fit_generator().动态生成数据(以及从磁盘读取)会产生额外的开销,因此在内存中的数据集上训练模型总是会更快.