Keras 快速解决OOM超内存的问题
更新时间：2020年06月11日 10:30:44 作者：silent56_th
这篇文章主要介绍了Keras 快速解决OOM超内存的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如果在Keras内部多次使用同一个Model，例如在不同的数据集上训练同一个模型进而得到结果，会存在内存泄露的问题。在运行几次循环之后，就会报错OOM。
解决方法是在每个代码后面接clear_session()函数，显示的关闭TFGraph，再重启。
详情参考 https://keras.io/backend/#backend-functions。
from keras import backend as K
K.clear_session()
补充知识：Keras多次加载模型进行测试时内存溢出的解决方法
在进行实验的过程中，保存了每个epoch的模型参数文件，需要验证每个保存的模型的效果，想到直接简单粗暴的手法，就是一个循环里加载模型并进行验证，但是导致随着加载的模型越来越多，速度越来越慢。
方法如下：在每次加载模型之前，清空模型占用的内存即可。
import tensorflow as tf from keras import backend as K K.clear_session() tf.reset_default_graph() ‘'‘加载模型位置'‘'
更多内容信息，可以参见官网介绍https://keras.io/api/utils/backend_utils/
以上这篇Keras 快速解决OOM超内存的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python实现员工管理系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-01-01
Python高阶函数extract与extractall使用实例探究
这篇文章主要为大家介绍了Python高阶函数extract与extractall使用实例探究，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2024-01-01
用Python和WordCloud绘制词云的实现方法(内附让字体清晰的秘笈)
这篇文章主要介绍了用Python和WordCloud绘制词云的实现方法(内附让字体清晰的秘笈),文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-01-01
Python 随机生成中文验证码的实例代码，需要的朋友可以参考一下2013-03-03
Matplotlib 文本和标注可以为数据和图形之间提供额外的信息，帮助观察者更好地理解数据和图形的含义，下面就将通过示例依次介绍文本和标注的常用使用方式2023-08-08
python实现三种字符串格式化方法(%、format、f-string)
本文主要介绍了python实现三种字符串格式化方法,主要包括%、format、f-string,具有一定的参考价值,感兴趣的可以了解一下2023-10-10
这篇文章主要介绍了python使用tkinter模块实现文件选择功能，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-06-06
这篇文章主要介绍了Python 如何修改程序默认时区，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-09-09
这篇文章主要介绍了python爬虫之场内ETF基金获取，ETF 是一种场内交易型基金，可以在盘中进行交易，交易性比场外基金强一点，下文基于python的相关资料展开，需要的小伙伴可以参考一下2022-05-05
这篇文章主要给大家介绍了关于Python对图像进行灰度处理的相关资料,图像灰度化是将一幅彩色图像转换为灰度化图像的过程,文中通过代码介绍的非常详细,需要的朋友可以参考下2023-11-11
最新评论