比如大家需要对某个keras的模型进行改进,但是必须使用预训练权重(毕竟不是每个人都有财力进行训练的),可以在对模型进行调整之后,使用以下函数
model.load_weights('filename.h5' , by_name = True, skip_mismatch = True)
代码的意义就是讲权重中相同名字的层载入,即使有部分结构微调,也可以保证权重加载成功
比如大家需要对某个keras的模型进行改进,但是必须使用预训练权重(毕竟不是每个人都有财力进行训练的),可以在对模型进行调整之后,使用以下函数
model.load_weights('filename.h5' , by_name = True, skip_mismatch = True)
代码的意义就是讲权重中相同名字的层载入,即使有部分结构微调,也可以保证权重加载成功