torchlosses / __init__.py
GenAIDevTOProd's picture
Upload folder using huggingface_hub
4c1e73e verified
raw
history blame contribute delete
228 Bytes
from .focal import FocalLoss
from .dice import DiceLoss
from .contrastive import ContrastiveLoss
from .triplet import TripletLoss
from .cosine import CosineEmbeddingLoss
from .huber import HuberLoss
from .kldiv import KLDivLoss