Градиенты для переменной TF2 не указаны

Question

Это моя потеря:

def custom_loss(input_mask):
   mask_value = K.cast(input_mask, dtype="float32")
   def masked_loss(y_true, y_pred):
      loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
      with tf.init_scope():
         losss = loss(y_true, y_pred) * mask_value
      return losss
  return masked_loss

Я использую его в компиляции модели

 masked_loss = custom_loss(input_mask)
 model.compile(loss=masked_loss, optimizer='adam', metrics=['accuracy'])

, но после попытки тренинга я получаю:

ValueError : Градиенты не указаны для любой переменной: бла бла все слои

Как я могу решить эту проблему?