预测时 内存一直在增加

[soulbyfeng]

你好，想请教下在预测时发现内存一直在涨，能帮忙解答下吗

[nl8590687]

是这样的，确实存在这种情况，这是因为使用TensorFlow后端的Keras深度学习框架存在内存泄漏的缘故。

[soulbyfeng]

谢谢楼主，我这边已经解决了，希望能给遇到同样问题的小伙伴一个思路 主要问题：预测时候调用k.ctc_decode会给图加默认的节点，导致图越来越来，内存暴增。 解决思路：理解函数含义，重写k.ctc_decode函数 输入（samples，feature_len,class_num)输出一个一维数组。 我的思路很简单，取输入数组中概率最大值的下表，删除默认值，就可以得到最终的输出，经验证和decode结果一样 代码如下： temp = base_pred.reshape(200, 1424) temp_arr = temp.argmax(1) arr_result=[] for k in range(200): # print(k) if temp_arr[k]!=1423: arr_result.append(temp_arr[k]) np_arr=np.array(arr_result)