mmpretrain/tests/test_models
zzc98 9506241f73
[Feature] Add arcface head. (#926)
* 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>
2022-11-02 17:45:33 +08:00
..
test_backbones [Feature] Support MobileViT backbone. (#1068) 2022-10-18 17:05:59 +08:00
test_utils [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
test_classifiers.py [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
test_heads.py [Feature] Add arcface head. (#926) 2022-11-02 17:45:33 +08:00
test_losses.py [Improve] Update benchmark scripts (#1028) 2022-09-20 15:50:21 +08:00
test_necks.py [Improve] Multiple fixes. (#957) 2022-08-09 15:58:08 +08:00
test_retrievers.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00