MIA-GCL / MVGRL / GCL2 / models / __init__.py
__init__.py
Raw
from .samplers import SameScaleSampler, CrossScaleSampler, get_sampler
from .contrast_model import SingleBranchContrast, DualBranchContrast, DualBranchContrast_mia,WithinEmbedContrast, BootstrapContrast


__all__ = [
    'SingleBranchContrast',
    'DualBranchContrast',
    'DualBranchContrast_mia',
    'WithinEmbedContrast',
    'BootstrapContrast',
    'SameScaleSampler',
    'CrossScaleSampler',
    'get_sampler'
]

classes = __all__