Keras加載自定義損失函數

在用Keras訓練好模型後，加載時出現如下異常：

ValueError: Unknown loss function:sample_loss

沒有找到自定義的損失函數。加載方法以下：

# 加載模型
self.model.set_weights(load_model(self.dirPath + str(self.fileName) + ".h5").get_weights())

須要調整加載代碼，來讓Keras找到自定義的模型

# 加載模型
self.model.set_weights(load_model(self.dirPath + str(self.fileName) + ".h5",custom_objects={"sample_loss":sample_loss}).get_weights())

這樣，Keras就能夠找到該函數了。