471 Commits

Author SHA1 Message Date
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
fangyixiao18
c9b86b8333 Use in-place operation to speed up EMA 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
5b8c1a8355 [Refactor]: Refactor config tutorial 2022-07-18 11:06:44 +08:00
fangyixiao18
5ea54a48c8 fix ut 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
ad382c2115 [Refactor] add unit test for deepcluster dataset and sampler 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
ceda0eed68 [Refactor] add unit test for part of model utils 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
YuanLiuuuuuu
0f10a4debe [Feature]: Add ut for backbones 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
62c909d3d2 [Refactor]: Rename optimizer wrapper constructor 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
301ae8ded2 [Refactor] Refactor memories 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
8d18f42c32 [Fix]: Fix UT 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0a2a811acf [Fix]: Fix lint 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
4fa194ae60 [Refactor]: Refactor random aug 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
7039be72d6 [Refactor]: Refactor PackSelfInputs 2022-07-18 11:06:44 +08:00