mmclassification/mmpretrain/models/utils
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
..
batch_augments Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
__init__.py [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
attention.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
batch_shuffle.py [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
channel_shuffle.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
clip_generator_helper.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00
data_preprocessor.py [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
ema.py [Refactor] Move and refactor utils from mmselfsup. (#1385) 2023-02-28 17:04:40 +08:00
embed.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
helpers.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
inverted_residual.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
layer_scale.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
make_divisible.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
norm.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
position_encoding.py [Refactor] Add necks, heads and losses for the self-supervised task. (#1376) 2023-02-28 10:05:00 +08:00
se_layer.py Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
vector_quantizer.py [Refactor] Add self-supervised backbones and target generators. (#1379) 2023-02-28 15:59:17 +08:00