loss = dice_loss + bce_loss