keras的load_model实现加载含有参数的自定义模型


Posted in Python onJune 22, 2020

网上的教程大多数是教大家如何加载自定义模型和函数,如下图

keras的load_model实现加载含有参数的自定义模型

这个SelfAttention层是在训练过程自己定义的一个class,但如果要加载这个自定义层,需要在load_model里添加custom_objects字典,这个自定义的类,不要用import ,最好是直接复制进再训练的模型中,这些是基本教程。

------------------分割线讲重点------------------

如果直接运行上面的代码,会出现一个init初始化错误,如下图,

keras的load_model实现加载含有参数的自定义模型

再来看看 这个SelfAttention 自定义的类的初始化

keras的load_model实现加载含有参数的自定义模型

这就说明再调用这个类的时候,输入的ch=256并不会初始化这个类,需要先自定义好初始化值,如下图

keras的load_model实现加载含有参数的自定义模型

调用方式不变

keras的load_model实现加载含有参数的自定义模型

这样问题就解决啦!

补充知识: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实现加载含有参数的自定义模型就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
把大数据数字口语化(python与js)两种实现
Feb 21 Python
python实现的简单FTP上传下载文件实例
Jun 30 Python
Python中文分词工具之结巴分词用法实例总结【经典案例】
Apr 15 Python
Python使用微信SDK实现的微信支付功能示例
Jun 30 Python
Python3 循环语句(for、while、break、range等)
Nov 20 Python
python实现验证码识别功能
Jun 07 Python
TensorFlow实现iris数据集线性回归
Sep 07 Python
Python实现将通信达.day文件读取为DataFrame
Dec 22 Python
Python turtle画图库&&画姓名实例
Jan 19 Python
解决python使用list()时总是报错的问题
May 05 Python
Java多线程实现四种方式原理详解
Jun 02 Python
Python中Cookies导出某站用户数据的方法
May 17 Python
python GUI模拟实现计算器
Jun 22 #Python
keras CNN卷积核可视化,热度图教程
Jun 22 #Python
python实现斗地主分牌洗牌
Jun 22 #Python
解决Keras使用GPU资源耗尽的问题
Jun 22 #Python
Keras - GPU ID 和显存占用设定步骤
Jun 22 #Python
Python 基于jwt实现认证机制流程解析
Jun 22 #Python
python中format函数如何使用
Jun 22 #Python
You might like
php获取操作系统语言代码
2013/11/04 PHP
php socket实现的聊天室代码分享
2014/08/16 PHP
php使用COPY函数更新配置文件的方法
2015/06/18 PHP
PHP+apc+ajax实现的ajax_upload上传进度条代码
2016/01/25 PHP
php 使用redis锁限制并发访问类示例
2016/11/02 PHP
js操作时间(年-月-日 时-分-秒 星期几)
2010/06/20 Javascript
js实现单一html页面两套css切换代码
2013/04/11 Javascript
网页禁用右键菜单和鼠标拖动选择方法小结
2015/02/25 Javascript
使用AngularJS编写较为优美的JavaScript代码指南
2015/06/19 Javascript
JavaScript如何自定义trim方法
2015/07/28 Javascript
JavaScript常用标签和方法总结
2015/09/01 Javascript
AngularJS入门教程之AngularJS指令
2016/04/18 Javascript
javascript 小数乘法结果错误的处理方法
2016/07/28 Javascript
js数字计算 误差问题的快速解决方法
2017/02/28 Javascript
基于vue实现多引擎搜索及关键字提示
2017/03/16 Javascript
jquery点赞功能实现代码 点个赞吧!
2020/05/29 jQuery
vue2.0使用v-for循环制作多级嵌套菜单栏
2018/06/25 Javascript
微信小程序实现多选功能
2018/11/04 Javascript
Electron 如何调用本地模块的方法
2019/02/01 Javascript
jquery树形插件zTree高级使用详解
2019/08/16 jQuery
vue实现滑动到底部加载更多效果
2020/10/27 Javascript
TF-IDF与余弦相似性的应用(二) 找出相似文章
2017/12/21 Python
Pytorch 实现自定义参数层的例子
2019/08/17 Python
双向RNN:bidirectional_dynamic_rnn()函数的使用详解
2020/01/20 Python
Django ORM filter() 的运用详解
2020/05/14 Python
Python实例教程之检索输出月份日历表
2020/12/16 Python
Sneaker Studio罗马尼亚网站:购买运动鞋
2018/11/04 全球购物
优秀的计算机专业求职信范文
2013/12/27 职场文书
护士感人事迹
2014/05/01 职场文书
关于保护环境的标语
2014/06/09 职场文书
2015元旦标语横幅
2014/12/09 职场文书
本科毕业论文致谢怎么写
2015/05/14 职场文书
课程设计感想范文
2015/08/11 职场文书
你会写报告?产品体验报告到底该怎么写?
2019/08/14 职场文书
2019年警察入党转正申请书最新范文
2019/09/03 职场文书
Nginx域名转发使用场景代码实例
2021/03/31 Servers