分类：超内存

2020
10-07

Keras 快速解决OOM超内存的问题

如果在Keras内部多次使用同一个Model，例如在不同的数据集上训练同一个模型进而得到结果，会存在内存泄露的问题。在运行几次循环之后，就会报错OOM。解决方法是在每个代码后面接clear_session()函数，显示的关闭TFGraph，再重启。详情参考https://keras.io/backend/#backend-functions。fromkerasimportbackendasKK.clear_session()补充知识：Keras多次加载模型进行测试时内存溢出的解决方法在进行实验的过程中，保存了每个e... 继续阅读 >