mmclassification/mmpretrain
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
..
apis [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor`. (#1380) 2023-02-28 16:31:42 +08:00
datasets [Fix] Fix value error while start training. (#1400) 2023-03-07 08:51:31 +08:00
engine [Feature] Implement layer-wise learning rate decay optimizer constructor. (#1399) 2023-03-07 17:30:39 +08:00
evaluation [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
models [Refactor] Use `out_type` to specify ViT-like backbone output. (#1408) 2023-03-09 11:02:58 +08:00
structures [Refactor] Move transforms in mmselfsup to mmpretrain. (#1396) 2023-03-03 15:01:11 +08:00
utils [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
visualization [Refactor] Refactor ClsDatasample to a union DataSample. (#1371) 2023-02-23 10:07:53 +08:00
__init__.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
registry.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
version.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00