保存模型:
model.save('...model.h5')
载入模型:
from keras.models import load_model
model = load_model('...model.h5')
#loss,accuracy = model.evaluate(x_test,y_test)
保存参数,载入参数
model.save_weights('...my_model_weights.h5')
model.load_weights('...my_model_weights.h5')
保存网络结构,载入网络结构
from keras.models import model_from_json
json_string=model.to_json()
# 载入网络结构
model=model_from_json(json_string)
keras 解决内存泄露问题:
详见: https://keras.io/backend/#backend-functions
在keras内部多次使用同一个model(在不同数据集上训练同一个模型而得到结果)
在运行几次循环后,会报错 OOM,解决方法是在每个代码后面接clear_session()函数,显示的关闭TFGraph,再重启
from keras import backend as K
K.clear_session()
在构建模型/加载模型前加上此句