105 Commits

Author SHA1 Message Date
Jiahao Xie
fc69e38009
Bump version to v0.6.0 (#199)
* [Feature] Add MoCo v3 (#194)

* [Feature] add position embedding function

* [Fature] modify nonlinear neck for vit backbone

* [Feature] add mocov3 head

* [Feature] modify cls_head for vit backbone

* [Feature] add ViT backbone

* [Feature] add mocov3 algorithm

* [Docs] revise BYOL hook docstring

* [Feature] add mocov3 vit small config files

* [Feature] add mocov3 vit small linear eval config files

* [Fix] solve conflict

* [Fix] add mmcls

* [Fix] fix docstring format

* [Fix] fix isort

* [Fix] add mmcls to runtime requirements

* [Feature] remove duplicated codes

* [Feature] add mocov3 related unit test

* [Feature] revise position embedding function

* [Feature] add UT codes

* [Docs] add README.md

* [Docs] add model links and results to model zoo

* [Docs] fix model links

* [Docs] add metafile

* [Docs] modify install.md and add mmcls requirements

* [Docs] modify description

* [Fix] using specific arch name `mocov3-small`  rather than general arch name `small`

* [Fix] add mmcls

* [Fix] fix arch name

* [Feature] change name to `MoCoV3`

* [Fix] fix unit test bug

* [Feature] change `BYOLHook` name to `MomentumUpdateHook`

* [Feature] change name to MoCoV3

* [Docs] modify description

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>

* [Docs] update model zoo results (#195)

* Bump version to v0.6.0 (#198)

* [Docs] update model zoo results

* Bump version to v0.6.0

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2022-02-02 11:16:06 +08:00
Yixiao Fang
21d631f528
[Enhance] speed up training (#181)
* [Enhance] speed up training

* [Enhance] add setup_multi_processes to test.py

* [Fix] fix lint
2022-01-25 15:50:28 +08:00
Ming-Yang Ho
aa1e65cee3
[Fix] fix a bug of showing kmeans losses (#182)
* fix a bug of showing kmeans losses

* fix a bug of showing kmeans losses

* isort

* fix typo (#180)

* add unit test

* remove gpu restrict

* restore gpu restrict

* bypass gpu by mock

* mock bypass

* mock bypass

* mock bypass

* mock bypass

* apply flake8

* apply flask8

* typo

* Upgrade isort (#184)

* Fix isort config

* Trigger commit hook

* Trigger commit hook

* Upgrade isort in ci and remove seed-isort-config

* Update test requirement

* Add unit test for build_dataset

* Fix test data

* Fix typo

* Use deepclustering dataset as test dataset config

* fix a bug of showing kmeans losses

* fix a bug of showing kmeans losses

* isort

* add unit test

* remove gpu restrict

* restore gpu restrict

* bypass gpu by mock

* mock bypass

* mock bypass

* mock bypass

* mock bypass

* apply flake8

* apply flask8

* typo

* apply isort

* remove session scope func

Co-authored-by: Uno Wu <st9007a@gmail.com>
2022-01-24 19:10:29 +08:00
Uno Wu
8369d6b0ce
Upgrade isort (#184)
* Fix isort config

* Trigger commit hook

* Trigger commit hook

* Upgrade isort in ci and remove seed-isort-config

* Update test requirement

* Add unit test for build_dataset

* Fix test data

* Fix typo

* Use deepclustering dataset as test dataset config
2022-01-24 15:22:31 +08:00
fangyixiao18
8f9ff74736 [Refactor] add UT codes 2021-12-15 19:07:01 +08:00