mmclassification/mmpretrain/models/selfsup
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
..
__init__.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
base.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
beit.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
cae.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
mae.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
maskfeat.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
milan.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
mixmim.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
mocov3.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
simclr.py [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
simmim.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00