所以我有一个设置,其中有一些数据并在其上训练Keras模型。一旦这样做,我将添加更多数据并再次训练模型。此过程将持续数次。但是最终,我的计算机说它的CPU RAM用完了。我怀疑Keras模型每次都在构建新图。我该如何检查?下面是我的重置方法。
def reset(self):
session = K.get_session()
for layer in self.model.layers:
if hasattr(layer, 'kernel_initializer'):
layer.kernel.initializer.run(session=session)
if hasattr(layer, 'bias_initializer') and layer.bias is not None:
layer.bias.initializer.run(session=session)
任何有关调试内存问题的帮助或技巧将不胜感激。我正在使用Keras 2.1.6