keras保存模型

一、

1.单纯保存权重：

model.save_weights( "my_model.hdf5" )

通过这种方式保存的模型加载的时候，必须先构建模型,然后编译模型：

my_model=model()
my_model.compile(optimizer=Adam(lr=1e-5))

2.最后才能加载权重：

my_model.load_weights("my_model.hdf5")

---

3.如果用到迁移学习之类的，hdf5文件和建立的模型不是完全一样，希望使用hdf5文件的部分层的权重，加载权重的时候可以通过设置by_name属性来完成：

my_model.load_weights('my_model.hdf55', by_name=True)

二、

1.保存模型的时候同时保存模型和权重：

my_model.save('my_model.hdf5') 

 2.这时候加载模型的话可以只用简单的代码加载模型和权重：

from keras.models import load_model
 
my_model = load_model('my_model.hdf5')