网络编程
位置:首页>> 网络编程>> Python编程>> 解决keras backend 越跑越慢问题

解决keras backend 越跑越慢问题

作者:MSTK  发布时间:2022-05-27 17:36:58 

标签:keras,backend

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

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

tf.Graph.finalize()

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

补充知识:win10下pytorch,tensorflow,keras+tf速度对比

采用GitHub上的代码

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


pytorchtensorflowkeras+tensorflow
version0.4.01.8.0Keras: 2.1.6 Tensorflow: 1.8.0
train time:1min 14s1min 9s1min 51s
evaluate time:378 ms9.4 s826 ms

来源:https://www.cnblogs.com/mstk/p/10632406.html

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com