16 lines
534 B
Python
16 lines
534 B
Python
from .backbones import * # noqa: F401,F403
|
|
from .builder import (build_backbone, build_model, build_head, build_loss)
|
|
from .byol import BYOL
|
|
from .heads import *
|
|
from .classification import Classification
|
|
from .deepcluster import DeepCluster
|
|
from .odc import ODC
|
|
from .necks import *
|
|
from .npid import NPID
|
|
from .memories import *
|
|
from .moco import MOCO
|
|
from .registry import (BACKBONES, MODELS, NECKS, MEMORIES, HEADS, LOSSES)
|
|
from .rotation_pred import RotationPred
|
|
from .relative_loc import RelativeLoc
|
|
from .simclr import SimCLR
|