keras模型保存和权重保存

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

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

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

模型权重的保存和读取:

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

model.load_weights('./weights.h5') # 由于这个只是保存了权重,所以在这句话之前还要重新定义网络

猜你喜欢

转载自www.cnblogs.com/rising-sun/p/12078464.html
今日推荐