编程 Python

解决keras backend 越跑越慢问题

Posted in Python onJune 18, 2020

Keras运行迭代一定代数以后，速度越来越慢，经检查是因为在循环迭代过程中增加了新的计算节点，导致计算节点越来越多，内存被占用完，速度变慢。

判断是否在循环迭代过程中增加了新的计算节点，可以用下面的语句：

tf.Graph.finalize()

如果增加了新的计算节点，就会报错，如果没有报错，说明没有增加计算节点。

补充知识：win10下pytorch，tensorflow，keras+tf速度对比

采用GitHub上的代码

运行类似vgg模型，在cifar10上训练，结果朋友torch与tensorflow速度相当，远远快过keras。

	pytorch	tensorflow	keras+tensorflow
version	0.4.0	1.8.0	Keras: 2.1.6 Tensorflow: 1.8.0
train time：	1min 14s	1min 9s	1min 51s
evaluate time：	378 ms	9.4 s	826 ms

以上这篇解决keras backend 越跑越慢问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

解决keras backend 越跑越慢问题

- Author -

MSTK

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐

python 判断自定义对象类型

Mar 21 Python

Python学习小技巧之列表项的推导式与过滤操作

May 20 Python

python使用Apriori算法进行关联性解析

Dec 21 Python

python实现顺序表的简单代码

Sep 28 Python

Python合并2个字典成1个新字典的方法(9种)

Dec 19 Python

kafka监控获取指定topic的消息总量示例

Dec 23 Python

flask 框架操作MySQL数据库简单示例

Feb 02 Python

Python Opencv轮廓常用操作代码实例解析

Sep 01 Python

解决PyCharm不在run输出运行结果而不是再Console里输出的问题

Sep 21 Python

使用python把xmind转换成excel测试用例的实现代码

Oct 12 Python

Django model class Meta原理解析

Nov 14 Python

Python如何利用pandas读取csv数据并绘图

Jul 07 Python

Python基于Twilio及腾讯云实现国际国内短信接口

Jun 18 #Python

详解Windows下PyCharm安装Numpy包及无法安装问题解决方案

Jun 18 #Python

解决Keras中Embedding层masking与Concatenate层不可调和的问题

Jun 18 #Python

Win10下用Anaconda安装TensorFlow(图文教程)

Jun 18 #Python

python中tab键是什么意思

Jun 18 #Python

python中可以声明变量类型吗

Jun 18 #Python

tensorflow之读取jpg图像长和宽实例

Jun 18 #Python