mmpretrain/tests
Ma Zerun dbf3df21a3
[Refactor] Use `out_type` to specify ViT-like backbone output. (#1408)
* [Refactor] Use  to specify ViT-like backbone output.

* Fix ClsBatchNormNeck

* Update mmpretrain/models/necks/mae_neck.py

---------

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2023-03-09 11:02:58 +08:00
..
data [Refactor] Update datasets (#1375) 2023-02-27 15:42:22 +08:00
test_apis [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor`. (#1380) 2023-02-28 16:31:42 +08:00
test_datasets [Refactor] Move transforms in mmselfsup to mmpretrain. (#1396) 2023-03-03 15:01:11 +08:00
test_engine [Feature] Implement layer-wise learning rate decay optimizer constructor. (#1399) 2023-03-07 17:30:39 +08:00
test_evaluation/test_metrics [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
test_models [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
test_structures [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
test_utils Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
test_visualizations [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
__init__.py [Feature] Support Multi-task. (#1229) 2022-12-30 10:36:00 +08:00