无法加载具有未知自定义 lambda 损失函数的 keras 模型答案

【问题标题】：can't load keras model with unknown custom lambda loss function无法加载具有未知自定义 lambda 损失函数的 keras 模型
【发布时间】：2020-08-01 21:30:09
【问题描述】：

我的模型看起来像

Keras 中的单输出多个损失函数：https://stackoverflow.com/a/51705573/9079093

model = Model(inputs=[sketch_inp, color_inp], outputs=disc_outputs)

opt = Adam(lr=learning_rate, beta_1=.5)



model.compile(loss=lambda y_true, y_pred : tf.keras.losses.binary_crossentropy(y_true, y_pred) + \
                                                 pixelLevelLoss_weight * pixelLevelLoss(y_true, y_pred) + \
                                                 totalVariationLoss_weight * totalVariationLoss(y_true, y_pred) + \
                                                 featureLevelLoss_weight * featureLevelLoss(y_true, y_pred),\
                    optimizer=opt)

保存模型后，我想加载它并完成训练，但我不知道如何使用此自定义损失函数加载它

【问题讨论】：

标签： python tensorflow keras loss

【解决方案1】：

在加载模型时，只需使用 cutom_objects 参数来传递损失。

如果您要加载的模型包含自定义层或其他自定义类或函数，您可以通过custom_objects 参数将它们传递给加载机制：

from keras.models import load_model
# Assuming your model includes instance of an "AttentionLayer" class
model = load_model('my_model.h5', custom_objects={'AttentionLayer': AttentionLayer})

【讨论】：