解决keras backend 越跑越慢问题
Keras运行迭代一定代数以后,速度越来越慢,经检查是因为在循环迭代过程中增加了新的计算节点,导致计算节点越来越多,内存被占用完,速度变慢。
判断是否在循环迭代过程中增加了新的计算节点,可以用下面的语句:
tf.Graph.finalize()
如果增加了新的计算节点,就会报错,如果没有报错,说明没有增加计算节点。
补充知识:win10下pytorch,tensorflow,keras+tf速度对比
采用GitHub上的代码
运行类似vgg模型,在cifar10上训练,结果朋友torch与tensorflow速度相当,远远快过keras。
以上这篇解决kerasbackend越跑越慢问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
热门推荐
10 财运虎年祝福语简短语
11 单亲爸爸春节祝福语简短
12 狗狗上厕所训练
13 植物种植祝福语简短
14 翻译考试祝福语朋友简短
15 个人周年庆简短祝福语
16 狗狗怎么训练
17 分别祝福语简短搞笑英文
18 生日广州男生祝福语简短