如何将训练好的网络进行保存,我们可以用pickle或cPickle来保存Keras模型,同时我们可以用下面的方法:
一、保存整个模型
model.save(filepath)
将Keras模型和权重保存在一个HDF5文件中,该文件将包含:
- 模型的结构
- 模型的权重
- 训练配置(损失函数,优化器,准确率等)
- 优化器的状态,以便于从上次训练中断的地方
前提是已经安装python的h5py包.
如何将训练好的网络进行保存,我们可以用pickle或cPickle来保存Keras模型,同时我们可以用下面的方法:
一、保存整个模型
model.save(filepath)
将Keras模型和权重保存在一个HDF5文件中,该文件将包含: