1. 程式人生 > >Keras 迴圈訓練模型跑資料時記憶體洩漏的問題解決辦法

Keras 迴圈訓練模型跑資料時記憶體洩漏的問題解決辦法

在使用完模型之後,新增這兩行程式碼即可清空之前model佔用的記憶體:

import tensorflow as tf
from keras import backend as K

K.clear_session()
tf.reset_default_graph()