十年网站开发经验 + 多家企业客户 + 靠谱的建站团队
量身定制 + 运营维护+专业推广+无忧售后,网站问题一站解决
这篇文章主要为大家展示了keras怎么实现权重保存和权重载入,内容简而易懂,希望大家可以学习一下,学习完之后肯定会有收获的,下面让小编带大家一起来看看吧。
如果需要全部权重载入,直接使用权重载入方式
model.save_weights('./weigths.h6')
model2.load_weights('./weigths.h6')
但是有时候你只需要载入部分权重
所以你可以这样操作
首先,为所有层命名,在层中直接加入方法 name='layer1'
第二,使用,将你不需要载入权重的值更改名字。
最后,载入权重。
x=BatchNormalization(axis=channel_axis,name='layer2')(x) model2.layers[-1].name='pred' model2.load_weights('./weigths.h6',by_name=True)
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。