from .contrastive_head import ContrastiveHead
from .cls_head import ClsHead
from .multi_cls_head import MultiClsHead