model保存与加载

常用深度学习模型训练权重汇总

https://blog.csdn.net/mago2015/article/details/84033460

模型的保存

1、使用model.save(filepath)将模型与权重保存在一个HDF5文件中，该文件主要包括：

• 模型结构；
• 权重；
• 训练配置；
• 优化器；

可以使用命令：

           keras.models.load_model(filepath)

重新实例化模型，

2. 保存模型权重

model.save_weight("train_model_weights.h5")

加载模型权重

model.load_weight('train_model_weigths.5',by_name=True)