使用Keras预处理时出现内存错误

时间:2018-10-29 02:49:57

标签: python tensorflow keras jupyter-notebook

我正在研究一个在Keras中使用CNN对狗的品种进行分类的项目。在预处理数据时,出现此内存错误:https://i.stack.imgur.com/e9TkG.png。我该怎么办?

1 个答案:

答案 0 :(得分:1)

您应该使用data generator

出现此错误是因为您的计算机没有足够的RAM来容纳所有数据。如果编写生成器,则可以获取所需的数据,然后可以使用model.fit_generator()。这样,您就可以获取大量数据,对其进行预处理,然后将其转发给模型,再获取另一批数据,对其进行预处理并转发给模型,等等。

您可以找到更多的here