KerasのEfficientNetでValueError: Unknown activation function:swish

KerasのEfficientNetで学習してモデルを保存し、load_modelで読み込んでpredictしようとした時に表題のエラー。

swishはKerasには元々存在しないカスタムオブジェクトなので、必要なモジュールを事前にimportしておく必要がある。

trainingと違ってpredictの時にはimportする必要がないと思っていたが、裏で必要になっていたようだ。

import efficientnet.keras

参考

ValueError: Unknown activation function:swish_activation - Stack Overflow

コメント

タイトルとURLをコピーしました