如何在Keras的自定义损失函数中使用经过训练的TensorFlow模型

Question

问题陈述

我想在自定义损失函数中使用预训练的TensorFlow模型来训练另一个模型，但收到错误消息。我在此代码中使用TensorFlow 2.3 Keras，其中名为'discriminator_model'的预训练模型已移交给custom_loss：

代码段

ID

错误消息

def custom_loss(discriminator_model):

    def loss_function(y_actual, y_pred):
        prediction_actual = discriminator_model.predict(y_actual)
        prediction_pred = discriminator_model.predict(y_pred)
        custom_loss = (prediction_pred - prediction_actual)**2
        return custom_loss

    return loss_function

...

    model.compile(optimizer=tf.keras.optimizers.Adam(lr_schedule), loss=custom_loss(discriminator_model), metrics='mse')
    generator_model.fit(DG_train, batch_size=batch_size, epochs=epochs)

是否可以在自定义损失函数中使用预训练的模型？我认为应该存在，因为这也是GAN背后的基本思想，对吗？但是如何？