強弱と高低

Python3をメインに

Keras2でカスタムレイヤー(自作の)の読み込みをする

Keras2でカスタムレイヤーを作った時

load_modelを単純にしようとすると,「そんなLayerは無い」と怒られる.

load_modelの引数を追加する

load_model(model_name, custom_objects={'AAA': AAA}) としてあげる.

FAQ - Keras DocumentationHandling custom layers (or other custom objects) in saved models に記載されている例のとおりにすると,

AttentionLayerを追加したとすると, load_model(model_name, custom_objects={'AttentionLayer': AttentionLayer}) としてあげるということ. 基本的に文字列を与える部分は, クラス名と同じになる.