Блог

Моя модель имеет два выходных слоя, с помощью которых я компилирую модель

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

 

MyLoss-это класс,который имеет функцию вызова(y_true, y_pred), а также у MyLoss есть инициализация, где он вызывает super().__init__(name=name)

Я пытаюсь загрузить эту сохраненную модель с помощью

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

Но когда я это делаю , я вижу ошибку ValueError: Unknown loss function:auto

Я не определил функцию автоматической потери