keras的load_model实现加载含有参数的自定义模型
作者:月下滑板鞋~ 时间:2021-03-05 08:18:56
网上的教程大多数是教大家如何加载自定义模型和函数,如下图
这个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即可解决加载问题。
来源:https://blog.csdn.net/qq_38855907/article/details/105472265
标签:keras,load,model,自定义,模型
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Python的加密模块md5、sha、crypt使用实例
2022-02-17 14:44:44
Python入门教程(十四)Python的集合
2022-11-29 21:21:09
![](https://img.aspxhome.com/file/2023/1/90531_0s.png)
Python一行代码解决矩阵旋转的问题
2022-03-08 12:44:11
PHP基于非递归算法实现先序、中序及后序遍历二叉树操作示例
2023-08-16 04:46:47
![](https://img.aspxhome.com/file/2023/7/55387_0s.jpg)
Python使用APScheduler实现定时任务过程解析
2023-01-23 19:20:35
![](https://img.aspxhome.com/file/2023/0/82640_0s.png)
Python通过DOM和SAX方式解析XML的应用实例分享
2023-10-15 10:46:32
Python发送http请求解析返回json的实例
2022-04-11 03:38:28
Tensorflow中使用tfrecord方式读取数据的方法
2023-12-05 01:01:18
![](https://img.aspxhome.com/file/2023/1/77001_0s.png)
ASP函数验证带小数点数字格式
2010-01-02 20:41:00
教你用FrontPage2003轻松布局网页
2008-02-24 17:05:00
![](https://img.aspxhome.com/file/UploadPic/20082/24/frontpage2003_79s.gif)
asp是什么格式 asp文件用什么打开
2020-06-30 16:04:48
Python爬取世界杯热搜榜并制作脚本自动发送信息到邮箱的过程
2021-01-21 20:35:53
![](https://img.aspxhome.com/file/2023/1/80401_0s.jpg)
python提效小工具之统计xmind用例数量(源码)
2021-11-26 21:42:44
![](https://img.aspxhome.com/file/2023/6/72246_0s.png)
HTML 5 胜出:XHTML2 宣告夭折
2009-07-12 15:23:00
常见Dreamweaver使用过程中的问题及解决办法
2011-03-17 16:16:00
python matplotlib中文显示参数设置解析
2022-11-18 14:11:41
![](https://img.aspxhome.com/file/2023/0/89410_0s.jpg)
实现div可编辑的常见方法
2007-11-06 12:02:00
ubuntu16.04制作vim和python3的开发环境
2022-08-11 00:11:07
Python嵌套函数,作用域与偏函数用法实例分析
2022-12-03 08:46:15
简单理解PHP的面向对象编程方式
2023-06-13 15:38:08