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_model model=load_model("weights.hdf5") This is error i am gettin...

コメント

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