Ma Zerun
|
c4ccae40db
|
[Docs] Update user guides docs and tools for MMPretrain. (#1429)
* [Docs] Update user guides docs and tools for MMPretrain.
* Fix UT
* Fix Chinese docs.
* Improve according to comments.
* Fix windows CI.
|
2023-03-27 14:32:26 +08:00 |
Yixiao Fang
|
08dc8c75d3
|
[Refactor] Add selfsup algorithms. (#1389)
* remove basehead
* add moco series
* add byol simclr simsiam
* add ut
* update configs
* add simsiam hook
* add and refactor beit
* update ut
* add cae
* update extract_feat
* refactor cae
* add mae
* refactor data preprocessor
* update heads
* add maskfeat
* add milan
* add simmim
* add mixmim
* fix lint
* fix ut
* fix lint
* add eva
* add densecl
* add barlowtwins
* add swav
* fix lint
* update readtherdocs rst
* update docs
* update
* Decrease UT memory usage
* Fix docstring
* update DALLEEncoder
* Update model docs
* refactor dalle encoder
* update docstring
* fix ut
* fix config error
* add val_cfg and test_cfg
* refactor clip generator
* fix lint
* pass check
* fix ut
* add lars
* update type of BEiT in configs
* Use MMEngine style momentum in EMA.
* apply mmpretrain solarize
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
|
2023-03-06 16:53:15 +08:00 |
|
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 |
|
Yixiao Fang
|
e453a45d31
|
[Refactor] Add self-supervised backbones and target generators. (#1379)
* add heads
* add losses
* fix
* remove mim head
* add modified backbones and target generators
* add unittest
* refactor caevit
* add window_size check
* fix lint
* apply new DataSample
* fix ut error
* update ut
* fix ut
* fix lint
* Update base modules.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
|
2023-02-28 15:59:17 +08:00 |
|
Yixiao Fang
|
63d9f27fde
|
[Refactor] Add necks, heads and losses for the self-supervised task. (#1376)
* add necks
* refactor linear neck
* rename simmim neck
* add heads
* add losses
* fix
* add unittest
* update
* update cae
* remove mim head
* update config
|
2023-02-28 10:05:00 +08:00 |
|
mzr1996
|
0979e78573
|
Rename the package name to `mmpretrain`.
|
2023-02-17 15:20:55 +08:00 |