Dassl.pytorch/dassl/modeling/head/__init__.py (2 lines of code) (raw):

from .build import build_head, HEAD_REGISTRY # isort:skip from .mlp import mlp