tensorflow2.0 - 自定义损失函数 TF 2.0 中的 Model.get_weights

Question

我正在设计一个自定义损失函数，我需要在损失函数中访问模型权重。

代码：

def my_loss(y_true, y_pred):
    model.get_weights()
    return K.sum(-(y_true * K.log(y_pred)))/batch_size + ((error2/num_conv)*scal_f)

但是当我尝试通过调用 get_weights() 访问损失函数中的权重时，它给了我以下错误

错误 =“无法在 Tensorflow 图形函数中获取值。”

标签： tensorflow2.0tf.keras

Answer 1

我在一些帮助下发现问题是默认情况下在 tf 2.0 中急切执行我只是将其关闭并且它有效