模型保存和读取(包括权重):
model.save('./model.h5')
from keras import models
model = models.load_model(./model.h5) # 读取之前不需要重新定义网络框架
模型权重的保存和读取:
model.save_weights('./weights.h5')
model.load_weights('./weights.h5') # 由于这个只是保存了权重,所以在这句话之前还要重新定义网络
模型保存和读取(包括权重):
model.save('./model.h5')
from keras import models
model = models.load_model(./model.h5) # 读取之前不需要重新定义网络框架
模型权重的保存和读取:
model.save_weights('./weights.h5')
model.load_weights('./weights.h5') # 由于这个只是保存了权重,所以在这句话之前还要重新定义网络