как создать пользовательскую функцию потерь для модели глубокого обучения

Question

Я работаю над моделью CNN для сегментации и хочу написать пользовательскую функцию потерь для модели

Я пытаюсь использовать tf.cond(), но она показывает

'NoneType' object cannot be interpreted as an integer.

Естьпонятия не имею, чтобы сделать это.bce_dice_loss является предыдущей функцией потерь и хорошо работает для модели

def own_loss(y_true, y_pred):
    y_true_f = K.flatten(y_true)
    a = tf.cast(tf.constant(10),'float32')
    return tf.cond(tf.reduce_sum(y_true_f) < a, 
                   lambda: a, 
                   lambda: bce_dice_loss(y_true, y_pred))