python - tensorflow 2中是否允许在自定义损失函数中调用模型？

Question

我正在尝试在 Tensorflow 2 中编写一个自定义损失函数来训练我的神经网络。我想知道是否允许从损失函数中调用模型（我正在训练）。使用这样的损失函数进行训练似乎没有问题，但我想确保在继续之前这样做不会有任何问题。

例子：

def custom_loss(model, dummy_data):
   def loss(y_true, y_pred):
      loss1 = tf.reduce_mean(tf.square(u_pred - y_true))
      loss2 = tf.reduce_mean(model(dummy_data))
      return loss1+0.5*loss2
   return loss

model = Model(input_tensor, output_tensor)
model.compile(loss=custom_loss(model, dummy_data), optimizer='adam')

任何意见将不胜感激。

标签： pythontensorflowkerasneural-networkloss-function