为什么keras中的model.fit函数会显着增加RAM内存？

Question

我使用open_memmap函数加载数据，它占用5GB RAM内存。然后，我编译具有参数的模型：89,268,608，它不占用任何RAM内存。我的批量大小目前为200，输入图像的形状为（300,54,3）。

我的问题是，当我在keras中调用model.fit函数时，我的RAM内存从5 GB增加到24GB。我的问题是为什么？

当我尝试使用不同的批处理大小时，什么都没有改变，仍然占用了23 GB的RAM？

如果有人可以解释我的情况，我将不胜感激，

谢谢！

Answer 1

Keras的fit方法可一次将所有数据加载到内存中，这意味着更改批处理大小不会影响它占用的RAM。看看使用fit_generator（专为大型数据集而设计）。