keras模型保存和权重保存

模型保存和读取(包括权重):python

model.save('./model.h5')

from keras import models
model = models.load_model(./model.h5) # 读取以前不须要从新定义网络框架

模型权重的保存和读取:网络

model.save_weights('./weights.h5')

model.load_weights('./weights.h5') # 因为这个只是保存了权重,因此在这句话以前还要从新定义网络
相关文章
相关标签/搜索