Dassl.pytorch/dassl/modeling/head/__init__.py

from .build import build_head, HEAD_REGISTRY  # isort:skip

from .mlp import mlp