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实现加载含有参数的自定义模型就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家介绍了Pandas字符串操作的各种方法及速度测试,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-08-08
这篇文章主要介绍了Python实现查找数据库最接近的数据,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
Python cookbook（数据结构与算法）从任意长度的可迭代对象中分解元素操作示例
这篇文章主要介绍了Python 数据结构与算法 从任意长度的可迭代象中分解元素操作,结合实例形式分析了Python使用*表达式针对可迭代对象的分解操作相关实现技巧,需要的朋友可以参考下2018-02-02
使用 Flask、Celery 和 Python 实现每月定时任务的步骤
下面给大家分享使用 Flask、Celery 和 Python 实现每月定时任务的步骤,本文分步骤结合脚本给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧2024-08-08
这是去年毕设做的一个Web漏洞扫描小工具，主要针对简单的SQL注入漏洞、SQL盲注和XSS漏洞。下文给大家介绍了使用说明和源代码，一起看看吧2016-10-10
Python CategoricalDtype自定义排序实现原理解析
这篇文章主要介绍了Python CategoricalDtype自定义排序实现原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
有很多小伙伴们都苦恼自己手速不够,今天特地整理了这篇文章,教你用Python实现一个打字测试工具来测试你的打字速度,文中有非常详细的代码示例,对想练手速的小伙伴们很有用哦,需要的朋友可以参考下2021-05-05
数据清洗是数据处理过程中至关重要的一部分,本文主要介绍了如何使用Python数据清洗库,具有一定的参考价值,感兴趣的可以了解一下2023-12-12
这篇文章主要介绍了自定义PyCharm快捷键的设置方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-05-05
这篇文章主要介绍了使用python实现简易五子棋小游戏,文中提供了三种实现方式,解决思路和部分实现代码,感兴趣的朋友可以参考下2023-03-03
最新评论