mmpretrain/tests/test_models
Yixiao Fang d80ec5a4b8
[Refactor] Refactor BEiT backbone and support v1/v2 inference. (#1144)
* refactor beit backbone

* use LinearClsHead

* fix mean and std value

* fix lint

* support inference if beit-v2

* update encoder layer and init

* update

* add ut

* add prepare_relative_position_bias_table function

* add cls_token

* fix lint

* add pos_embed check

* update metafile and readme

* update weights link

* update link of weights

* update metafile

* update

* update docstrings

* update according to review

* rename readme

* update docstring

* fix lint
2022-11-29 12:56:33 +08:00
..
test_backbones [Refactor] Refactor BEiT backbone and support v1/v2 inference. (#1144) 2022-11-29 12:56:33 +08:00
test_utils [Improve] Speed up data preprocessor. (#1064) 2022-10-17 17:08:18 +08:00
test_classifiers.py [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. (#1102) 2022-11-10 14:56:19 +08:00
test_heads.py [Enhance] Enhance ArcFaceClsHead. (#1181) 2022-11-21 18:10:39 +08:00
test_losses.py [Improve] Update benchmark scripts (#1028) 2022-09-20 15:50:21 +08:00
test_necks.py [Feature] Add reduction for neck (#978) 2022-11-04 15:44:37 +08:00
test_retrievers.py [Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 2022-11-02 17:43:56 +08:00