__init__.py 139 Bytes
Newer Older
lishj6's avatar
lishj6 committed
1
2
3
4
from .cross_entropy_loss import CrossEntropyLoss
from .focal_loss import CustomFocalLoss

__all__ = ['CrossEntropyLoss', 'CustomFocalLoss']