解决keras backend 越跑越慢问题

 更新时间:2020年06月18日 16:42:42   作者:MSTK  
这篇文章主要介绍了解决keras backend 越跑越慢问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

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 越跑越慢问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 使用python对excle和json互相转换的示例

    使用python对excle和json互相转换的示例

    今天小编就为大家分享一篇使用python对excle和json互相转换的示例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-10-10
  • 使用pandas实现筛选出指定列值所对应的行

    使用pandas实现筛选出指定列值所对应的行

    这篇文章主要介绍了使用pandas实现筛选出指定列值所对应的行,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • 用smtplib和email封装python发送邮件模块类分享

    用smtplib和email封装python发送邮件模块类分享

    本文针对发邮件相关的操作进行了封装,包括发送文本、HTML、带附件的邮件,使用Python发邮件,主要用到smtplib和email两个模块,需要的朋友可以参考下
    2014-02-02
  • python使用tcp实现局域网内文件传输

    python使用tcp实现局域网内文件传输

    这篇文章主要介绍了python使用tcp实现局域网内文件传输,文件包括文本,图片,视频等,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-07-07
  • python绘制云雨图raincloud plot

    python绘制云雨图raincloud plot

    这篇文章主要介绍了python绘制云雨图raincloud plot,Raincloud的Python实现是一个名为PtitPrince的包,它写在seaborn之上,这是一个Python绘图库,用于从pandas数据帧中获取漂亮的绘图
    2022-08-08
  • Python发送http请求解析返回json的实例

    Python发送http请求解析返回json的实例

    下面小编就为大家分享一篇Python发送http请求解析返回json的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-03-03
  • 基于Python实现评论区抽奖功能详解

    基于Python实现评论区抽奖功能详解

    这篇文章主要为大家介绍了基于Python如何实现评论区抽奖的功能,文章的示例代码讲解详细,对我们学习Python有一定帮助,需要的朋友可以学习一下
    2021-12-12
  • Python&Matlab实现蚂蚁群算法求解最短路径问题的示例

    Python&Matlab实现蚂蚁群算法求解最短路径问题的示例

    本文主要介绍了Python&Matlab实现蚂蚁群算法求解最短路径问题的示例,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-03-03
  • Python绘制散点图之可视化神器pyecharts

    Python绘制散点图之可视化神器pyecharts

    这篇文章主要介绍了Python绘制散点图之可视化神器pyecharts,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-07-07
  • 全面理解python命名空间字典

    全面理解python命名空间字典

    本文主要介绍了全面理解python命名空间字典,python的命名空间由字典实现,属性为键,对象为值,通过属性找到对象,下面就来具体了解一下,感兴趣的可以了解一下
    2023-12-12

最新评论