from models.heads.classification import ClassificationHead
from models.heads.projection import ProjectionHead

__all__ = ["ClassificationHead", "ProjectionHead"]