mmclassification/mmpretrain/models/selfsup
Yixiao Fang a1cfe888e2
[Feature] Support SparK. (#1531)
* add spark configs

* fix configs

* remove repeat aug

* add module codes

* support lr layer decay of resnet

* update

* fix lint

* add metafile and readme

* fix lint

* add models and logs

* refactor codes

* fix lint

* update model rst

* update name

* add docstring

* add ut

* fix lint

---------

Co-authored-by: Ma Zerun <mzr1996@163.com>
2023-06-19 11:27:50 +08:00
..
__init__.py [Feature] Support SparK. (#1531) 2023-06-19 11:27:50 +08:00
barlowtwins.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
base.py [Feature] Implement layer-wise learning rate decay optimizer constructor. (#1399) 2023-03-07 17:30:39 +08:00
beit.py [Fix] Fix ddp bugs caused by `out_type`. (#1570) 2023-05-17 17:32:10 +08:00
byol.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
cae.py [Fix] Fix ddp bugs caused by `out_type`. (#1570) 2023-05-17 17:32:10 +08:00
densecl.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
eva.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
itpn.py [Feature] Support iTPN and HiViT (#1584) 2023-05-26 12:08:34 +08:00
mae.py [Feature] Support iTPN and HiViT (#1584) 2023-05-26 12:08:34 +08:00
maskfeat.py [Fix] Fix ddp bugs caused by `out_type`. (#1570) 2023-05-17 17:32:10 +08:00
milan.py [Fix] Fix clip generator init bug (#1518) 2023-04-25 19:35:09 +08:00
mixmim.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
moco.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
mocov3.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
simclr.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
simmim.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
simsiam.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00
spark.py [Feature] Support SparK. (#1531) 2023-06-19 11:27:50 +08:00
swav.py [Refactor] Add selfsup algorithms. (#1389) 2023-03-06 16:53:15 +08:00