mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
a1cfe888e2
* add spark configs * fix configs * remove repeat aug * add module codes * support lr layer decay of resnet * update * fix lint * add metafile and readme * fix lint * add models and logs * refactor codes * fix lint * update model rst * update name * add docstring * add ut * fix lint --------- Co-authored-by: Ma Zerun <mzr1996@163.com>
57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .barlowtwins import BarlowTwins
|
|
from .base import BaseSelfSupervisor
|
|
from .beit import VQKD, BEiT, BEiTPretrainViT
|
|
from .byol import BYOL
|
|
from .cae import CAE, CAEPretrainViT, DALLEEncoder
|
|
from .densecl import DenseCL
|
|
from .eva import EVA
|
|
from .itpn import iTPN, iTPNHiViT
|
|
from .mae import MAE, MAEHiViT, MAEViT
|
|
from .maskfeat import HOGGenerator, MaskFeat, MaskFeatViT
|
|
from .milan import MILAN, CLIPGenerator, MILANViT
|
|
from .mixmim import MixMIM, MixMIMPretrainTransformer
|
|
from .moco import MoCo
|
|
from .mocov3 import MoCoV3, MoCoV3ViT
|
|
from .simclr import SimCLR
|
|
from .simmim import SimMIM, SimMIMSwinTransformer
|
|
from .simsiam import SimSiam
|
|
from .spark import SparK
|
|
from .swav import SwAV
|
|
|
|
__all__ = [
|
|
'BaseSelfSupervisor',
|
|
'BEiTPretrainViT',
|
|
'VQKD',
|
|
'CAEPretrainViT',
|
|
'DALLEEncoder',
|
|
'MAEViT',
|
|
'MAEHiViT',
|
|
'iTPNHiViT',
|
|
'iTPN',
|
|
'HOGGenerator',
|
|
'MaskFeatViT',
|
|
'CLIPGenerator',
|
|
'MILANViT',
|
|
'MixMIMPretrainTransformer',
|
|
'MoCoV3ViT',
|
|
'SimMIMSwinTransformer',
|
|
'MoCo',
|
|
'MoCoV3',
|
|
'BYOL',
|
|
'SimCLR',
|
|
'SimSiam',
|
|
'BEiT',
|
|
'CAE',
|
|
'MAE',
|
|
'MaskFeat',
|
|
'MILAN',
|
|
'MixMIM',
|
|
'SimMIM',
|
|
'EVA',
|
|
'DenseCL',
|
|
'BarlowTwins',
|
|
'SwAV',
|
|
'SparK',
|
|
]
|