elektronn3.modules.loss module

Loss functions

elektronn3.modules.loss.dice_loss(probs, target, weight=1.0, eps=0.0001, smooth=0.0)[source]
elektronn3.modules.loss.global_average_pooling(inp)[source]
Return type

Tensor

elektronn3.modules.loss.norpf_dice_loss(probs, target, weight=1.0, class_weight=1.0)[source]