from .cls_head import ClsHead
from .linear_head import LinearClsHead
from .multi_label_head import MultiLabelClsHead
from .multi_label_linear_head import MultiLabelLinearClsHead
from .vision_transformer_head import VisionTransformerClsHead
from .tau_norm_linear_head import TauNormLinearClsHead

__all__ = [
    'ClsHead', 'LinearClsHead', 'MultiLabelClsHead', 'MultiLabelLinearClsHead',
    'VisionTransformerClsHead', 'TauNormLinearClsHead'
]
