如果在Keras内部多次使用同一个Model,例如在不同的数据集上训练同一个模型进而得到结果,会存在内存泄露的问题。在运行几次循环之后,就会报错OOM。解决方法是在每个代码后面接clear_session()函数,显示的关闭TFGraph,再重启。详情参考 https://keras.io/backend/#backend-functions。
from keras import backend as K
K.clear_session()
如果在Keras内部多次使用同一个Model,例如在不同的数据集上训练同一个模型进而得到结果,会存在内存泄露的问题。在运行几次循环之后,就会报错OOM。解决方法是在每个代码后面接clear_session()函数,显示的关闭TFGraph,再重启。详情参考 https://keras.io/backend/#backend-functions。
from keras import backend as K
K.clear_session()