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实现加载含有参数的自定义模型就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持软件开发网。
您可能感兴趣的文章:使用Keras加载含有自定义层或函数的模型操作keras自定义损失函数并且模型加载的写法介绍keras读取训练好的模型参数并把参数赋值给其它模型详解在Keras中实现保存和加载权重及模型结构
相关文章
Dianne
2020-09-19
Rebecca
2021-05-19
Hope
2021-03-01
Naomi
2020-03-18
Gretchen
2020-04-28
Oceana
2021-06-02
Malina
2023-07-19
Bunny
2023-07-20
Willow
2023-07-20
Lassie
2023-07-20
Carly
2023-07-20
Irma
2023-07-20
Posy
2023-07-20
Crystal
2023-07-20
Rachel
2023-07-20
Viridis
2023-07-20
Zahara
2023-07-20
Lida
2023-07-21
Bea
2023-07-21