mmclassification/mmpretrain
Yixiao Fang c9670173aa
[Refactor] Move and refactor utils from mmselfsup. (#1385)
* add heads

* add losses

* fix

* remove mim head

* add modified backbones and target generators

* fix lint

* fix lint

* add heads

* add losses

* fix

* add data preprocessor from mmselfsup

* add ut for data prepocessor

* add GatherLayer

* add ema

* add batch shuffle

* add misc

* fix lint

* update

* update docstring
2023-02-28 17:04:40 +08:00
..
apis [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor`. (#1380) 2023-02-28 16:31:42 +08:00
datasets [Refactor] Update datasets (#1375) 2023-02-27 15:42:22 +08:00
engine
evaluation [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
models [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
structures
utils [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
visualization
__init__.py
registry.py
version.py