* feat: add arcface head * feat: add arcface head * update arcface * refactor archface head * update archface head * update archface head * use torch.cat instead of torch.hstack to fix ci Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> |
||
---|---|---|
.. | ||
backbones | ||
classifiers | ||
heads | ||
losses | ||
necks | ||
retrievers | ||
utils | ||
__init__.py | ||
builder.py |