【keras】保存和加载神经网络模型

        一旦你利用Keras完成了训练,你可以将你的网络保存在HDF5里面。当然,你需要先安装 h5py。HDF5 格式非常适合存储大量的数据,并从 numpy 处理这些数据。比如,我们可以轻松的将存储在磁盘上的多TB数据集进行切片,就好像他们是真正的 numpy 数组一样。你还可以将多个数据集存储在单个文件中,遍历他们或者查看 .shape 和 .dtype 属性。

1.保存模型

        如果你要保存训练好的权重,那么你可以直接使用 save_weights:

model.save_weights("my_model.h5")

2.加载模型

model.load_weights("my_model_weights.h5")

参考自:VGG16学习笔记

猜你喜欢

转载自blog.csdn.net/zyxhangiian123456789/article/details/87617776