Commit 507aa382 authored by qubvel's avatar qubvel

Fix dice loss custom objects

parent 1b62620c
......@@ -92,7 +92,7 @@ def cce_dice_loss(gt, pr, cce_weight=1., class_weights=1., smooth=SMOOTH, per_im
# Update custom objects
get_custom_objects().update({
'jaccard_loss': jaccard_loss,
'bce_jaccard_loss': bce_jaccard_loss,
'cce_jaccard_loss': cce_jaccard_loss,
'dice_loss': dice_loss,
'bce_dice_loss': bce_dice_loss,
'cce_dice_loss': cce_dice_loss,
})
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment