我正在研究一个在Keras中使用CNN对狗的品种进行分类的项目。在预处理数据时,出现此内存错误:https://i.stack.imgur.com/e9TkG.png。我该怎么办?
答案 0 :(得分:1)
您应该使用data generator
。
出现此错误是因为您的计算机没有足够的RAM来容纳所有数据。如果编写生成器,则可以获取所需的数据,然后可以使用model.fit_generator()
。这样,您就可以获取大量数据,对其进行预处理,然后将其转发给模型,再获取另一批数据,对其进行预处理并转发给模型,等等。
您可以找到更多的here