keras自定义层对自定义参数进行保存

问题描述

在使用keras的过程中，有时会在自定义层中定义自己的参数，然而调用 json_model = model.to_json() 后发现自定义的参数并没有保存到模型中，造成模型加载时报错。

解决方案

需要在该层中实现 get_config(self) 方法，将自定义的参数保存到一个map中。聚义可以参考本人的《Keras 实现对bert编码的封装实现》中的具体写法。