在进行实验的过程中,保存了每个epoch的模型参数文件,需要验证每个保存的模型的效果,想到直接简单粗暴的手法,就是一个循环里加载模型并进行验证,但是导致随着加载的模型越来越多,速度越来越慢。
方法如下:在每次加载模型之前,清空模型占用的内存即可。
import tensorflow as tf
from keras import backend as K
K.clear_session()
tf.reset_default_graph()
‘’‘加载模型位置’‘’
更多内容信息,可以参见官网介绍https://keras.io/api/utils/backend_utils/