2021
07-04
07-04
keras的get_value运行越来越慢的解决方案
keras深度学习框架中get_value函数运行越来越慢,内存消耗越来越大问题问题描述如上图所示,经过时间和内存消耗跟踪测试,发现是keras.backend.get_value()函数导致的程序越来越慢,而且严重的造成内存泄露;查看该函数内部实现,发现一个主要核心是x.eval(session=get_session()),该语句可能是导致内存泄露和运行慢的核心语句;根据查看一些博文得到了运行得越来越慢的原因:该x.eval函数会添加新的节点到tf的图中;而这也导致...
继续阅读 >