没有合适的资源?快使用搜索试试~ 我知道了~
首页keras的load_model实现加载含有参数的自定义模型
keras的load_model实现加载含有参数的自定义模型
9 下载量 77 浏览量
更新于2023-03-16
评论
收藏 92KB PDF 举报
网上的教程大多数是教大家如何加载自定义模型和函数,如下图 这个SelfAttention层是在训练过程自己定义的一个class,但如果要加载这个自定义层,需要在load_model里添加custom_objects字典,这个自定义的类,不要用import ,最好是直接复制进再训练的模型中,这些是基本教程。 ——————分割线讲重点—————— 如果直接运行上面的代码,会出现一个init初始化错误,如下图, 再来看看 这个SelfAttention 自定义的类的初始化 这就说明再调用这个类的时候,输入的ch=256并不会初始化这个类,需要先自定义好初始化值,如下图 调用方式不变 这样问
资源详情
资源评论
资源推荐
keras的的load_model实现加载含有参数的自定义模型实现加载含有参数的自定义模型
网上的教程大多数是教大家如何加载自定义模型和函数,如下图
这个SelfAttention层是在训练过程自己定义的一个class,但如果要加载这个自定义层,需要在load_model里添加
custom_objects字典,这个自定义的类,不要用import ,最好是直接复制进再训练的模型中,这些是基本教程。
——————分割线讲重点——————
如果直接运行上面的代码,会出现一个init初始化错误,如下图,
再来看看 这个SelfAttention 自定义的类的初始化
这就说明再调用这个类的时候,输入的ch=256并不会初始化这个类,需要先自定义好初始化值,如下图
调用方式不变
这样问题就解决啦!
补充知识:补充知识:keras load model的时候,报错的时候,报错(‘Keyword argument not understood:’, u’******’)如何解决如何解决
由于keras不同版本的API有变化,因此在一个keras版本下训练的模型在另一个keras版本下加载时,可能会出现诸如
(‘Keyword argument not understood:’, u’data_format’)等报错。
通过打开*.h5文件,查看该模型训练所用环境,再安装该环境即可解决报错。
查看Keras Model所用的Keras环境的方法
import h5py
f = h5py.File('Model.h5', 'r')
print(f.attrs.get('keras_version'))
根据输出的keras版本安装对应版本的keras即可解决加载问题。
以上这篇keras的load_model实现加载含有参数的自定义模型就是小编分享给大家的全部内容了,希望能给大家一个参考,也
希望大家多多支持软件开发网。
您可能感兴趣的文章您可能感兴趣的文章:使用Keras加载含有自定义层或函数的模型操作keras自定义损失函数并且模型加载的写法介绍keras读取
训练好的模型参数并把参数赋值给其它模型详解在Keras中实现保存和加载权重及模型结构
weixin_38675815
- 粉丝: 3
- 资源: 889
上传资源 快速赚钱
- 我的内容管理 收起
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
会员权益专享
最新资源
- zigbee-cluster-library-specification
- JSBSim Reference Manual
- c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf
- 建筑供配电系统相关课件.pptx
- 企业管理规章制度及管理模式.doc
- vb打开摄像头.doc
- 云计算-可信计算中认证协议改进方案.pdf
- [详细完整版]单片机编程4.ppt
- c语言常用算法.pdf
- c++经典程序代码大全.pdf
- 单片机数字时钟资料.doc
- 11项目管理前沿1.0.pptx
- 基于ssm的“魅力”繁峙宣传网站的设计与实现论文.doc
- 智慧交通综合解决方案.pptx
- 建筑防潮设计-PowerPointPresentati.pptx
- SPC统计过程控制程序.pptx
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论0