在多个损失角点上训练具有单个输出的模型

时间:2019-11-04 11:18:20

标签: python tensorflow keras deep-learning loss-function

我正在使用keras构建图像分割模型,我想在多种损失函数上训练我的模型。我见过this链接,但是由于这种损失函数非常复杂,因此我正在寻找一种更简单直接的解决方案。有人可以告诉我如何建立一个具有单一输出且在喀拉拉邦中造成多个损失的模型。

1 个答案:

答案 0 :(得分:0)

您可以将多个损失与加权损失一并用于输出,这是损失乘以权重的总和。创建您的自定义损失,该损失将返回其他损失的总和,并将其传递给model.compile。有一个示例here