日々精進

新しく学んだことを書き留めていきます

tf.keras.load_modelでモデルを読み込むと「ValueError: Unknown loss function」エラー

学習時にcustom loss functionを使った場合、load時も同じ関数をcustom_objectとして渡してやらないといけないらしい。 ただ、再学習するのではなく推論するためにmodelをloadするのであれば以下のようにcompile=Falseを指定してやればOK。

model = load_model("lc_model.h5", compile=False)

参考:

stackoverflow.com