keras的load_model实现加载含有参数的自定义模型
更新时间：2020年06月22日 11:25:56 作者：月下滑板鞋~
这篇文章主要介绍了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实现加载含有参数的自定义模型就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python中if和elif的区别，下面文章将举多个例子围绕if和eli的相关资料展开内容,需要的朋友可以参考一下，希望对你有所帮助2021-11-11
Python使用win32com模块实现数据库表结构自动生成word表格的方法
这篇文章主要介绍了Python使用win32com模块实现数据库表结构自动生成word表格的方法,结合实例形式分析了win32com模块下载、连接mysql、查询获取表结构以及使用win32com生成word表格的相关操作技巧,需要的朋友可以参考下2018-07-07
这篇文章主要介绍了python3 解决requests出错重试的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-04-04
本文主要讲诉了字符串的定义，变量链接到字符串，以及对字符串的简单操作，都是些非常基础的东西，适合零基础的Pythoner学习，觉得有用的话，多鼓鼓掌吧2014-09-09
这篇文章主要介绍了基于python实现蓝牙通信代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
这篇文章主要介绍了pandas 如何分割字符的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
这篇文章主要介绍了详解pygame捕获键盘事件的两种方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-04-04
这篇文章主要为大家详细介绍了python创建股票的时间序列可视化分析，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-03-03
这篇文章主要介绍了Python partial函数原理及用法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
这篇文章主要介绍了Python抓取Discuz!用户名脚本代码，有需要的朋友可以参考一下2013-12-12
最新评论