分类：内存不释放

2020
10-08

解决Keras中循环使用K.ctc_decode内存不释放的问题

如下一段代码，在多次调用了K.ctc_decode时，会发现程序占用的内存会越来越高，执行速度越来越慢。data=generator(...)model=init_model(...)foriinrange(NUM):x,y=next(data)_y=model.predict(x)shape=_y.shapeinput_length=np.ones(shape[0])*shape[1]ctc_decode=K.ctc_decode(_y,input_length)[0][0]out=K.get_value(ctc_decode)原因每次执行ctc_decode时都会向计算图中添加一个节点，这样... 继续阅读 >