Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
__init__.py		__init__.py
angular.py		angular.py
base.py		base.py
detection.py		detection.py
dice_jaccard.py		dice_jaccard.py
focal.py		focal.py
functional.py		functional.py
hinge.py		hinge.py
huber.py		huber.py
kld.py		kld.py
lovasz.py		lovasz.py
smooth.py		smooth.py
vgg_loss.py		vgg_loss.py
wing_loss.py		wing_loss.py

README.md

Collection of losses

All new losses should be inherited from .base.Loss class to ensure that they can be combined in a nice way.

Example of combining multiple losses

from pytorch_tools import losses
custom_loss = losses.BinaryDiceLoss() * 0.5 + losses.BinaryFocalLoss() * 5