MIA-GCL / GRACE / GCL2 / losses / __init__.py
__init__.py
Raw
from .jsd import JSD, DebiasedJSD, HardnessJSD
from .vicreg import VICReg
from .infonce import InfoNCE, InfoNCESP, DebiasedInfoNCE, HardnessInfoNCE
from .triplet import TripletMargin, TripletMarginSP
from .bootstrap import BootstrapLatent
from .barlow_twins import BarlowTwins
from .losses import Loss

__all__ = [
    'Loss',
    'InfoNCE',
    'InfoNCESP',
    'DebiasedInfoNCE',
    'HardnessInfoNCE',
    'JSD',
    'DebiasedJSD',
    'HardnessJSD',
    'TripletMargin',
    'TripletMarginSP',
    'VICReg',
    'BarlowTwins'
]

classes = __all__