15 lines
525 B
Python
15 lines
525 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .cls_head import ClsHead
|
|
from .contrastive_head import ContrastiveHead
|
|
from .latent_pred_head import LatentClsHead, LatentPredictHead
|
|
from .mae_head import MAEFinetuneHead, MAEPretrainHead
|
|
from .mocov3_head import MoCoV3Head
|
|
from .multi_cls_head import MultiClsHead
|
|
from .swav_head import SwAVHead
|
|
|
|
__all__ = [
|
|
'ContrastiveHead', 'ClsHead', 'LatentPredictHead', 'LatentClsHead',
|
|
'MultiClsHead', 'SwAVHead', 'MAEFinetuneHead', 'MAEPretrainHead',
|
|
'MoCoV3Head'
|
|
]
|