Commit Graph

550 Commits (41747f73c774757ac7585f16f9f9a90960f7e9e1)
 

Author SHA1 Message Date
Wangbo Zhao(黑色枷锁) 9ddef4f7d9
[Refactor] Modify documents of classification, detection and segmentation (#426)
* modify config files

* modify benchmark docs

* Update classification.md

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify doc

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify benchmark docs

* modify benchmark doc
2022-08-29 14:31:06 +08:00
Jiahao Xie e7525f7163
[Refactor] refine docstring for algorithms and memories (#432) 2022-08-24 15:31:44 +08:00
Yixiao Fang 3c51e86d44
[Fix] Fix ut (#422)
* add register all modules

* add scope
2022-08-19 14:36:26 +08:00
Songyang Zhang c176fdad6e
[Fix] update pip requirement (#427)
* [Fix] fix sklearn bug in requirements

* [Fix] fix lint
2022-08-19 13:58:50 +08:00
Yixiao Fang 25ed606ef2
[Refactor] rename config files, change `fp16` to `amp` (#423) 2022-08-19 13:38:37 +08:00
Yixiao Fang 805bbba93f
[Feature] Add benchmark scripts (#424)
* add benchmark scripts

* update metafile

* remove date time of work_dir

* parse ckpt path

* refactor pattern
2022-08-19 13:36:01 +08:00
RenQin adbcd176b6
[Fix]: fix docstring of visualization (#417) 2022-08-18 20:17:29 +08:00
Yixiao Fang 8de714fa76
[Refactor] Update import dependency (#428) 2022-08-18 15:36:58 +08:00
Yuan Liu 24c7255104
[Feature]: Add missing docstring for dataset (#414)
* [Feature]: Add missing docstring for dataset

* [Fix]: Fix lint

* [Fix]: Fix docstring
2022-08-17 14:28:31 +08:00
Yixiao Fang d3a487e0b9
[Refactor] Refactor docs directory (#419)
* refactor directory

* modify titles

* fix lint

* update index.rst

* update

* fix typo

* update

* fix typo

* update model zoo

* update index.rst

* fix typo

* fix typo
2022-08-17 12:06:41 +08:00
RenQin 83b1a33bed
[Refactor]: Rename SelfSupLocalVisualizer to SelfSupVisualizer (#392)
* [Refactor]: rename SelfSupLocalVisualizer to SelfSupVisualizer

* [Fix]: fix example errors in docstring
2022-08-15 16:01:15 +08:00
Wangbo Zhao(黑色枷锁) d2f490f515
[Refactor] Refactor benchmark config files (#399)
* modify config files

* modify config files

* modify config files

* modify config files

* modify config files

* modify config files

* modify config files

* modify configs

* modify configs

* modify config files
2022-08-14 18:21:12 +08:00
Yixiao Fang 8b104c9844
[Refactor] Refactor tools folder (#408) 2022-08-09 11:34:54 +08:00
Yixiao Fang 0ea07c0750
[Refactor] Refactor hooks to support non-dist training (#397)
* apply mmengine.dist instead of torch.dist

* appy get_model to densecl_hook

* fix bug to pass ut

* update typehint
2022-08-08 15:01:47 +08:00
Yuan Liu a703ba2fcb
[Refactor]: Refactor layer wise lr decay (#375)
* [Refactor]: Refactor layer wise lr decay

* [Fix]: Remove is_dcn_module and add log

* [Fix]: Fix lint

* [Fix]: Check relative pos bias table is not decayed

* [Fix]: Fix lint

* [Fix]: Fix UT
2022-08-05 17:38:53 +08:00
Yixiao Fang 045b1fde8e
[Refactor] Apply get_current_instance instead of get_root_logger (#394) 2022-08-04 20:26:09 +08:00
Yixiao Fang c8cf491c4c
[Refactor] Update version check rule (#393) 2022-08-04 17:12:15 +08:00
RenQin c7b4a8323a
[Docs]: Add comments to indicate which linear evaluation setting belogs to barlowtwins (#390) 2022-08-01 19:57:56 +08:00
Yuan Liu 5d6133d60c
[Refactor]: Remove MIM ViT related code (#387)
* [Refactor]: Delete mim vit and related files

* [Fix]: Remove mim-vit from __init__.py
2022-08-01 14:14:03 +08:00
Yixiao Fang ba50b7450b
[Refactor] Rename mmselfsup.data to mmselfsup.structures (#385) 2022-07-30 16:36:48 +08:00
Yuan Liu f39d1c19bf
[Fix]: Add custom import and use model config from mmcls (#373)
* [Fix]: Add custom import and use model config from mmcls

* [Fix]: Use custom import from base
2022-07-25 16:47:12 +08:00
Yixiao Fang f86828fe5a
fix custom import error (#371) 2022-07-25 15:47:53 +08:00
Yixiao Fang 067fbce0b5
[CI] Add circleci (#363)
* add circleci

* fix linting

* check version of torchvision

* fix torch dependency bug

* using torch.stack instead of vstack

* comment github actions temporarily

* pass docs

* update docs requirements
2022-07-25 14:11:57 +08:00
Yuan Liu 10d9539f67
[Feature]: Add reconstruction loss (#365)
* [Feature]: Add pixel reconstruction loss

* [Fix]: Fix lint
2022-07-22 17:28:50 +08:00
YuanLiuuuuuu 353e3f7c58 [Feature]: Add auto resume and amp 2022-07-21 14:32:33 +08:00
YuanLiuuuuuu d7fd0c0fd8 [Fix]: Delete redundant seed 2022-07-21 14:32:33 +08:00
Yixiao Fang 9582c6b1ec
fix args error (#362) 2022-07-21 10:36:39 +08:00
Yixiao Fang c3593472f4
Update Version (#350)
* update version information

* remove mmengine
2022-07-20 17:58:38 +08:00
Yuan Liu 05036a544d
[Refactor]: Refactor MAE (#349)
* [Refactor]: Refactor dataset config

* [Refactor]: Refactor mae model config

* [Refactor]: mae config 400e

* [Refactor]: Refactor mae 400e config fp16

* [Refactor]: Refactor mae

* [Refactor]: Refactor mae vit

* [Refactor]: Refactor mae head

* [Refactor]: Refactor mae neck

* [Fix]: Fix UT

* [Refactor]: Delete mc

* [Fix]: Use MODELS registry

* [Fix]: Delete mae ft and linear head

* [Fix]: Fix the train epochs in config

* [Fix]: Use MODELS from mmself.registry

* [Refactor]: Delete param in super

* [Fix]: Fix docstring

* [Fix]: Fix adamw_coslr-200e_in1k.py steps

* [Fix]: Use disk

* [Fix]: Change name
2022-07-20 17:24:37 +08:00
Yixiao Fang d4ee6e5c2c
[Refactor] Add backend option for ColorJitter (#353)
* add backend option for ColorJitter

* revise docstring
2022-07-19 19:00:29 +08:00
Yixiao Fang c7101c7648
[Enhance] support non-dist training of simclr (#352)
* support non-dist training of simclr

* revise ut
2022-07-19 14:21:26 +08:00
Yixiao Fang af3b261dce [Enhance] support non-dist training of mocov3 (#346)
* support non-dist training of mocov3

* revise docstring
2022-07-18 11:06:44 +08:00
Yixiao Fang 210ea79316 fix import error (#348) 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 4cb6aa3895 [Refactor] refactor package 2022-07-18 11:06:44 +08:00
xiejiahao.vendor f4c19381c9 [Refactor] refactor dc and odc 2022-07-18 11:06:44 +08:00
renqin 4e28ee7ff4 [Feature]: add local_visualizer 2022-07-18 11:06:44 +08:00
renqin 775364cf11 [Refactor]: refactor extract.py and the configs 2022-07-18 11:06:44 +08:00
fangyixiao.vendor d390a03a47 [Refactor] support to train moco in non-dist environment 2022-07-18 11:06:44 +08:00
renqin 4c5c4b88f4 [Refactor]: refactor npid algorithm 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 3e5d18ea83 [Fix]: Fix layer-wise decay lr scheduler 2022-07-18 11:06:44 +08:00
renqin dfa4d180df [Refactor]: refactor swav algorithm 2022-07-18 11:06:44 +08:00
renqin 5f778aa552 [Refactor]: refactor densecl algorithm 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu d6dfa9fe40 [Fix]: Move CAE data preprocessor to data preprocessor file and fix UT 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 8910743c6e [Fix]: Fix multi-head config 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 1e16016b27 [Refactor] refactor byol 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 29c6b26ee0 [Refactor] refactor tsne tool 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu acf001d554 [Refactor]: Refactor CAE 2022-07-18 11:06:44 +08:00
renqin a648611cc5 [Refactor] refactor simsiam 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 61fdc30c59 [Refactor] refactor simclr 2022-07-18 11:06:44 +08:00
fangyixiao.vendor c9a532e1c0 [Refactor] refactor mocov3 benchmark 2022-07-18 11:06:44 +08:00