多次重新训练Keras模型会导致内存不足错误

时间:2018-07-06 23:30:45

标签: python memory keras

所以我有一个设置,其中有一些数据并在其上训练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

0 个答案:

没有答案