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

悠悠 2023-02-20 02:09 39阅读 0赞

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

更多python视频教程请到菜鸟教程https://www.piaodoo.com/

如果在Keras内部多次使用同一个Model，例如在不同的数据集上训练同一个模型进而得到结果，会存在内存泄露的问题。在运行几次循环之后，就会报错OOM。

解决方法是在每个代码后面接clear\_session()函数，显示的关闭TFGraph，再重启。

详情参考 [https://keras.io/backend/\#backend-functions。][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/][https_keras.io_backend_backend-functions]

以上这篇Keras 快速解决OOM超内存的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多

茂名论坛https://www.hnthzk.com/

化州橘红http://www.sfkyty.com/

茂名论坛http://www.nrso.net/

源码搜藏网http://www.cntkd.net/

茂名市高级技工学校(茂名一技)http://www.szsyby.net/

茂名一技https://www.mmbbu.com/

[https_keras.io_backend_backend-functions]: https://keras.io/api/utils/backend_utils/

发表评论取消回复

表情：

评论列表（有 0 条评论，39人围观）

还没有评论，来说两句吧...

相关阅读

相关超快速定位OOM

想主动dump出JVM的内存，有挺多方式，但不管哪种方式，主动dump内存会引发STW，请择时操作。即通过arthas提供的命令heapdump主动dump出JVM的内存...

曾经终败给现在/ 2024年05月23日 22:22/ 0 赞/ 103 阅读

相关什么是OOM？如何快速定位并解决OOM？

解决OOM问题通常需要综合考虑应用程序本身的逻辑、资源管理、内存配置和代码优化等方面。定位和解决OOM问题需要仔细分析和排查，可以结合监控工具、堆转储文件和内存分析工具等...

矫情吗；*/ 2024年05月23日 22:21/ 0 赞/ 94 阅读

相关 04 虚拟内存、物理内存

虚拟内存的思想/作用每个程序拥有自己的地址空间，这个空间被分割成多块，每一块称作一页，每一页有连续的地址范围，这些页被映射到物理内存，但不需要映射到连续的物理内存

深藏阁楼爱情的钟/ 2023年03月01日 12:54/ 0 赞/ 69 阅读

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

Keras 快速解决OOM超内存的问题更多python视频教程请到菜鸟教程https://www.piaodoo.com/ 如果在Keras内部多次使用同一个Model，

悠悠/ 2023年02月20日 02:09/ 0 赞/ 40 阅读

相关快速串讲---JVM内存的区域划分

目的快速定位JVM内存泄漏或者溢出等问题。面试基础题，加分项。 ![20210426091155761.png][] 程序计数器（Program C

绝地灬酷狼/ 2023年01月16日 13:28/ 0 赞/ 149 阅读

相关物理内存、虚拟内存、buffers、cached、共享内存、swap

物理内存：实际使用的内存；虚拟内存：虚拟内存是操作系统内核为了对进程地址空间进行管理（process address space management）而精心

迷南。/ 2023年01月05日 14:29/ 0 赞/ 474 阅读

相关 JS中的栈内存、堆内存

先来看这样一个问题： > const定义的值能改么？答案是部分能改，部分不能改。const定义的基本类型不能改变，但是定义的对象是可以通过修改对象属性等方法来改变的。例如

蔚落/ 2022年12月23日 15:27/ 0 赞/ 273 阅读

相关快速读取进程内存

快速读取进程内存（摘自《Debug Hacks》，略做修改）转载自[http://www.kankanews.com/ICkengine/archives/3763.s

妖狐艹你老母/ 2022年10月02日 04:44/ 0 赞/ 247 阅读

相关基础：如何快速找出系统内存的问题

内存性能指标 > 内存使用情况已用内存：已经使用的内存剩余内存：还没使用的内存共享内存：共享内存是通过tmpfs实现的，所以它的大小也就是tmp

我会带着你远行/ 2022年09月17日 03:19/ 0 赞/ 190 阅读

相关内存

内存分为RAM和ROM： RAM，是断电就会丢失数据，又分为SRAM和DRAM。 SRAM访问速度快，主要是作为cpu的内存，dram主要是作为硬盘，隔断时间需要自刷新一

「爱情、让人受尽委屈。」/ 2022年03月29日 08:52/ 0 赞/ 290 阅读