Commit Graph

122 Commits (0c873137d021fc7774169b91f545d98be0fa3ae5)

Author SHA1 Message Date
Yixiao Fang 337e49e304 [Refactor] refactor Metafile format (#478)
* update metafile

* update format

* update metafile

* update mae
2022-10-12 19:35:44 +08:00
Yuan Liu 088d4c8285 [Fix]: Fix mae config name (#498) 2022-10-12 19:35:44 +08:00
Yixiao Fang f9a85146fa [Refactor] Update results of mocov3-r50 (#496)
* add ft config

* update mocov3 r50 config

* set last_bn of head to False

* data sp

* imagenet config

* update

* update

* change backend to disk

* update model zoo

* update readme

* update metafile
2022-10-12 19:35:44 +08:00
Yixiao Fang 2dacc14d4a [Fix]fix classification configs (#488) 2022-10-12 19:35:44 +08:00
Yuan Liu 14afb8e302 [Feature]: Full support of SimMIM (#487)
* [Fix]: Refine simmim-224-ft config

* [Feature]: Add simmim-b 800e pt config

* [Feature]: Add simmim large

* [Feature]: Add ceph

* [Fix]: Fix min lr and add pin_memory

* [Feature]: Add more simmim res in readme and model_zoo

* [Fix]: Refine lr

* [Fix]: Delete file client
2022-10-12 19:35:44 +08:00
Yuan Liu 1905569713 [Feature]: Update MAE README (#483)
* [Feature]: Update MAE README

* [Feature]: Add ViT-H to readme and model_zoo
2022-10-12 19:35:44 +08:00
Yuan Liu 776d64b7cd
[Fix]: Refine simmim-224-ft config (#473) 2022-09-07 15:18:27 +08:00
Yixiao Fang 6eebff6982
[Feature] Add mocov3 and simclr configs (#450)
* add mocov3 vit-b linear eval config

* add mocov3 vit-b cfg for pretrain

* update channel weights

* add mocov3 finetune config

* add mocov3 r50 config

* update comments

* update cfg

* add mocov3-r50 300e and 1000e configs

* update configs

* add 800e config

* refactor model configs

* refactor arch zoo format

* add mocov3 vit large

* update mocov3 vit-l config

* fix config error

* rename config

* add grad clip

* update config

* update

* rename the configs

* fix config

* add simclr config

* update config
2022-08-31 19:44:58 +08:00
Yuan Liu 0521abf354
[Fix]: Fix some bugs related to MAE and CAE (#430)
* [Fix]: Do not decay relative pos bias

* [Feature]: MAE ft benchmark config

* [Fix]: Fix mae benchmark config

* [Refactor]: Delete mc

* [Fix]: Add comment to config

* [Feature]: Add MAE linprob config

* [Feature]: Add ClsBatchNormNeck

* [Feature]: Add mc

* [Fix]: Fix custom import

* [Feature]: Add mae linear prob sgd optim

* [Feature]: Add petrel

* [Feature]: Add ceph and mc

* [Feature]: Add mae large

* [Feature]: Add MAE ViT-large config

* [Fix]: Fix MAE ft and pt config

* [Fix]: Fix ViT-L lr scheduler bug

* [Fix]: Change the ViT-large ft config name

* [Fix]: Change mc to petrel in vit benchmark

* [Feature]: Add mae huge config

* [Fix]: Fix ViT huge 100e pretrain setting

* [Feature]: Add ViT large ft

* [Feature]: Use ceph

* [Fix]: Fix MAE ViT-L 800e lr schedule

* [Feature]: Set mae vit-l start factor to 1e-9

* [Fix]: Delete drop_after_pos

* [Feature]: Create MAE ViT-h 1600e, remove 100e

* [Fix]: Fix cae ft config

* [Fix]: Set output cls token to False in mae ft

* [Fix]: Set drop path rate to 0.1 in mae large ft

* [Fix]: Use linear head in mae ft config

* [Fix]: Fix lint

* [Feature]: Add mae huge ft

* [Feature]: Add vit-h 448 ft

* [Refactor]: Change fp16 to amp in config name

* [Fix]: Use disk file client
2022-08-31 15:47:36 +08:00
Jiahao Xie b23765fce7
[Refactor] refactor knn (#420) 2022-08-31 13:32:03 +08:00
Yuan Liu 20488d01b4
[Refactor]: Refactor data flow (#429)
* [Refactor]: Refactor data flow

* [Fix]: Change data sample to data samples

* [Fix]: Change batch_inputs to inputs

* [Fix]: Fix lint and UT

* [Fix]: Fix UT

* [Fix]: Fix lint

* [Fix]: Fix docstring

* [Fix]: Fix UT

* [Refactor]: Add assert in data preprocessor

* [Fix]: Fix lint
2022-08-30 11:34:04 +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 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
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
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
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
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
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 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 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
YuanLiuuuuuu 35e0988527 [Refactor]: Refactor base model and create a sample for MAE 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu ded3dc9640 [Refactor]: Refacor scheduler and change scheduler to param_scheduler 2022-07-18 11:06:44 +08:00
renqin 2e1b978bb2 [Refactor]: refactor dataset config and transforms 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu 4fa194ae60 [Refactor]: Refactor random aug 2022-07-18 11:06:44 +08:00
renqin e34159e6bc [Feature]: add Places205 dataset 2022-07-18 11:06:44 +08:00
fangyixiao.vendor c39277b216 [Feature] apply EMA from mmengine and add CosineEMA 2022-07-18 11:06:44 +08:00
fangyixiao.vendor 09c721a33e [Refactor] refactor sampler and clean redundant codes 2022-07-18 11:06:44 +08:00