0

我的模型有两个输出层我正在使用它来编译模型

model.compile(
            optimizer="Adam",
            loss=[MyLoss(foo1, name="my_loss1"), MyLoss(foo2,name="my_loss2")],
            loss_weights=[0.5, 0.5],
        )

MyLoss 是一个类,它有一个 call(y_true,y_pred) 函数,而且 MyLoss在它调用的地方有initsuper().__init__(name=name)

我正在尝试使用加载此保存的模型

tf.keras.models.load_model(path, custom_objects={"MyLoss": MyLoss, "my_loss1": MyLoss, "my_loss2": MyLoss})

但是当我这样做时,我看到了错误ValueError: Unknown loss function:auto

我还没有定义自动损失函数

4

0 回答 0