keras自定義層對自定義參數進行保存

問題描述

在使用keras的過程中，有時會在自定義層中定義自己的參數，然而調用 json_model = model.to_json() 後發現自定義的參數並沒有保存到模型中，造成模型加載時報錯。

解決方案

需要在該層中實現 get_config(self) 方法，將自定義的參數保存到一個map中。聚義可以參考本人的《Keras 實現對bert編碼的封裝實現》中的具體寫法。