keras的load_model实现加载含有参数的自定义模型
作者：月下滑板鞋~ 发布时间：2021-03-05 08:18: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即可解决加载问题。
来源：https://blog.csdn.net/qq_38855907/article/details/105472265
0
投稿猜你喜欢
本文实例讲述了Python实现的对本地host127.0.0.1主机进行扫描端口功能。分享给大家供大家参考，具体如下：前面一篇《Python- 一、导入excel文件和相关库import pandas;import matplotlib;from pandas.tools.plott
- asp学习入门经验介绍，本文初步介绍了初学asp的一些相关知识，如VBScript语法简介，循环控制语句的使用，asp数据库的简单操作查询，
- 一、Python输出print是python输出的关键字，默认是输出内容后换行。如果不想换行，需要在变量末尾加上 end=&quo
- 一、思路往MySQL中插入1000000条数据只花了6秒钟！关键点：1.使用PreparedStatement对象2.rewriteBatc
- 微信小程序官方给了十分详细的登陆时序图，当然为了安全着想，应该加上签名加密。微信小程序端1).调用wx.login获取 code 。 2).
- Python json 错误xx is not JSON serializable解决办法在使用json的时候经常会遇到xxx
- asp无组件上传VBS编写的大家见的多了，这个是纯javascript实现的上传,原来unicode可以解决读取位置的问题，这次真的是纯JS
- 字符串转list数组str = '1,2,3'arr = str.split(',')gpu_ids分配na
- 线性回归属于监督学习，因此方法和监督学习应该是一样的，先给定一个训练集，根据这个训练集学习出一个线性函数，然后测试这个函数训练的好不好(即此
- 一、系统环境yum update升级以后的系统版本为[root@yl-web yl]# cat /etc/redhat-release Ce
- 本文从简单的例子入手，从打包文件去分析以下三个问题：webpack打包文件是怎样的？如何做到兼容各大模块化方案的？webpack3带来的新特
- panic源码解读前言本文是在go version go1.13.15 darwin/amd64上进行的panic的作用panic能够改变程
- 在查询语句中使用 NOLOCK 和 READPAST 处理一个数据库死锁的异常时候，其中一个建议就是使用 NOLOCK 或者 READPAS
- 今天来说说鄙人对input输入框在处理上的细节处理和心得，其实制作一个符合CSS标准、FF/IE7/IE6等主流浏览器全兼容、符合用户体验的
- 最近在做学院的选课系统时，在分页上被卡壳了一下，因为需要用到排序，所以不能像以前一样用一个自动递增的字段作为主键，然后仅仅是对这个主键来做统
- 首先要作出决策，你是否想要使用最新的开发版本或最终的稳定版本。在MySQL开发过程中，同时存在多个发布系列，每个发布处在成熟度的不同阶段：&
- web框架是什么?web开发框架是一组工具，同时也提供了非常多的资源，供软件开发人员构建和管理网站、提供web服务、编写web应用程序。它是
- 目的是为了检测出采集数据中的异常值。所以很明确，这种情况下的簇为2：正常数据和异常数据两大类1、安装相应的库import matplotli
- 一、anaconda的安装首先，下载安装包。Anaconda的下载方式有两种：通过官网下载，选择适合自己的电脑版本的安装包。https://
