自定义损失函数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（）访问损失函数中的权重时，会出现以下错误

Error =“无法在Tensorflow图函数中获取值。”

Answer 1

我在一些帮助下找到了问题，默认情况下该问题渴望在tf 2.0中执行，我只是将其关闭并且有效