如何为深度学习模型创建自定义损失函数

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))