Commit Graph

485 Commits (045b1fde8ea45d07ac4d94715d491eb61fe59ce8)
 

Author SHA1 Message Date
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
YuanLiuuuuuu 7b89e971e3 [Feature]: Add Swin 192 benchmark 2022-07-18 11:06:44 +08:00
fangyixiao.vendor bc199203a0 [Refactor] refactor rotation pred 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 87ed42aaeb [Refactor] refactor barlowtwins 2022-07-18 11:06:44 +08:00
renqin 7528fbcbe8 [Refactor]: refactor simsiam algorithm 2022-07-18 11:06:44 +08:00
fangyixiao.vendor cd04872c95 [Refactor] refactor classification benchmark, ResNet related config 2022-07-18 11:06:44 +08:00
fangyixiao.vendor da564e5b02 [Refactor] Refactor extractor and collect according to new BaseModel 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 408b9fb187 [Fix]: Fix layer wise learning rate decay optim wrapper 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 5a84e88609 [Refactor] refactor mocov3 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu bcc4576ace [Refactor]: Refactor relative loc 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 5ba17adb23 [Refactor]: Refactor SimMIM 2022-07-18 11:06:44 +08:00
fangyixiao.vendor b9df061e7f [Refactor] refactor moco according to new basemodel 2022-07-18 11:06:44 +08:00
fangyixiao.vendor df8c204d75 [Refactor] refactor hooks and add ut 2022-07-18 11:06:44 +08:00
fangyixiao.vendor f78fe71794 [Refactor] refactor swav hook to control gradient 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 35e0988527 [Refactor]: Refactor base model and create a sample for MAE 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 962f9b9752 [Refactor] refactor multi_cls_head and clean cls algorithm 2022-07-18 11:06:44 +08:00