keras的load_model实现加载含有参数的自定义模型

网上的教程大多数是教大家如何加载自定义模型和函数，如下图 这个SelfAttention层是在训练过程自己定义的一个class，但如果要加载这个自定义层，需要在load_model里添加custom_objects字典，这个自定义的类，不要用import ，最好是直接复制进再训练的模型中，这些是基本教程。 ——————分割线讲重点—————— 如果直接运行上面的代码，会出现一个init初始化错误，如下图， 再来看看 这个SelfAttention 自定义的类的初始化 这就说明再调用这个类的时候，输入的ch=256并不会初始化这个类，需要先自定义好初始化值，如下图 调用方式不变 这样问