【发布时间】:2018-10-29 02:49:56
【问题描述】:
我正在做一个项目,在 Keras 中使用 CNN 对犬种进行分类。在预处理数据时,我收到此内存错误:https://i.stack.imgur.com/e9TkG.png。我该怎么办?
【问题讨论】:
-
你能显示
path_to_tensor的代码吗?
标签: python tensorflow keras jupyter-notebook
我正在做一个项目,在 Keras 中使用 CNN 对犬种进行分类。在预处理数据时,我收到此内存错误:https://i.stack.imgur.com/e9TkG.png。我该怎么办?
【问题讨论】:
path_to_tensor的代码吗?
标签: python tensorflow keras jupyter-notebook
您应该使用data generator。
您收到此错误是因为您的计算机没有足够的 RAM 来容纳其上的所有数据。如果您编写生成器,那么您可以获得尽可能多的数据,然后您可以使用model.fit_generator()。有了这个,你可以抓取一些数据,对其进行预处理,并将其转发给模型,抓取另一批数据,对其进行预处理并转发给模型等等。
您可以了解更多here
【讨论】: