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
I am trying to load the save weights using keras load_model().from keras.models import load_modelmodel=load_model("weigh...

コメント

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