分类：权重保存

2020
09-29

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... 继续阅读 >