__init__.py 185 Bytes
Newer Older
Jiangmiao Pang's avatar
Jiangmiao Pang committed
1
2
3
4
5
from .cross_entropy_loss import CrossEntropyLoss
from .focal_loss import FocalLoss
from .smooth_l1_loss import SmoothL1Loss

__all__ = ['CrossEntropyLoss', 'FocalLoss', 'SmoothL1Loss']