keras的load_model如何加载含有参数的自定义模型

网上的教程大多数是教大家如何加载自定义模型和函数，如下图

这个SelfAttention层是在训练过程自己定义的一个class，但如果要加载这个自定义层，需要在load_model里添加custom_objects字典，这个自定义的类，不要用import ，最好是直接复制进再训练的模型中，这些是基本教程。

----------------------------------------分割线讲重点------------------

如果直接运行上面的代码，会出现一个init初始化错误，如下图，

再来看看 这个SelfAttention 自定义的类的初始化

这就说明再调用这个类的时候，输入的ch=256并不会初始化这个类，需要先自定义好初始化值，如下图

 

调用方式不变

这样问题就解决啦！

如有问题欢迎下方留言或私信我。