tensorflow2.3+ kears tf.keras.models.load_model载人模型，模型ValueError: Unknown loss function: define_loss

最新推荐文章于 2024-08-19 22:42:18 发布

Daisy和她的单程车票

最新推荐文章于 2024-08-19 22:42:18 发布

阅读量860

点赞数 1

分类专栏： tensorflow2.0 文章标签： tensorflow

本文链接：https://blog.csdn.net/u012513618/article/details/110543945

版权

在TensorFlow 2.3中遇到使用`tf.keras.models.load_model`加载包含自定义损失函数的模型时，可能会出现ValueError。本文提供了两种解决方案：1) 使用`custom_objects`参数指定自定义损失函数；2) 设置`compile=False`，以忽略损失函数错误，仅用于推理。参考自Stack Overflow上的相关讨论。

摘要由CSDN通过智能技术生成

自定义损失函数载入模型失效解决方案

def sum_loss(inputs):
    ctr_true, ctr_pred, ctcvr_true, ctcvr_pred = inputs
    ctr_loss = tf.keras.losses.binary_crossentropy(y_true=ctr_true, y_pred=ctr_pred)
    ctcvr_loss =