keras 权重保存和权重载入

如果需要全部权重载入，直接使用权重载入方式
model.save_weights('./weigths.h5') model2.load_weights('./weigths.h5')
 

但是有时候你只需要载入部分权重

所以你可以这样操作

首先，为所有层命名，在层中直接加入方法　name='layer1'

第二，使用，将你不需要载入权重的值更改名字。

最后，载入权重。
x=BatchNormalization(axis=channel_axis,name='layer2')(x)
model2.layers[-1].name='pred' model2.load_weights('./weigths.h5',by_name=True)
上面的代码是对应的操作，这里我除了最后一层，其他层我都加载了权重，记住，by_name 必须赋值为True 这样才能够按照名称对应赋值权重。

注意：两个模型结构必须一样，不然可能出问题

• « 上一篇：基于Django的乐观锁与悲观锁解决订单并发问题的一点浅见
• » 下一篇：【通俗理解】显著性检验，T-test，P-value