编程 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编程实现数学运算求一元二次方程的实根算法示例

Apr 02 Python

Flask框架URL管理操作示例【基于@app.route】

Jul 23 Python

Python中应该使用%还是format来格式化字符串

Sep 25 Python

解决python3 安装完Pycurl在import pycurl时报错的问题

Oct 15 Python

用xpath获取指定标签下的所有text的实例

Jan 02 Python

解决pycharm的Python console不能调试当前程序的问题

Jan 20 Python

python构造函数init实例方法解析

Jan 19 Python

python sorted函数原理解析及练习

Feb 10 Python

Python如何使用paramiko模块连接linux

Mar 18 Python

Python文件时间操作步骤代码详解

Apr 13 Python

keras的load_model实现加载含有参数的自定义模型

Jun 22 Python

python绘制趋势图的示例

Sep 17 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