Keras2でカスタムレイヤー(自作の)の読み込みをする
Keras2でカスタムレイヤーを作った時
load_modelを単純にしようとすると,「そんなLayerは無い」と怒られる.
load_modelの引数を追加する
load_model(model_name, custom_objects={'AAA': AAA})
としてあげる.
FAQ - Keras Documentation
の Handling custom layers (or other custom objects) in saved models
に記載されている例のとおりにすると,
AttentionLayerを追加したとすると,
load_model(model_name, custom_objects={'AttentionLayer': AttentionLayer})
としてあげるということ. 基本的に文字列を与える部分は, クラス名と同じになる.