正文

keras模型保存和权重保存

IP云
IP云V管理员/0 阅读
08/03
小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

模型保存和读取（包括权重）：

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

模型权重的保存和读取：

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