535 Commits

Author SHA1 Message Date
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
renqin
e34159e6bc [Feature]: add Places205 dataset 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
1176808264 [Refactor] refactor utils 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
YuanLiuuuuuu
17136bf935 [Feature]: Add pseudo_label in selfsupdatasample 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
a10a6bd64f [Refactor]: Refactor layer wise decay learning rate optimizer constructor 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
2f2813ecd4 [Refactor]: Refactor ODC 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0baac605d1 [Refactor]: Refactor NPID 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
YuanLiuuuuuu
0c88e8dceb [Refactor]: Refactor loss 2022-07-18 11:06:44 +08:00
fangyixiao18
1f69aa84f7 add 'set_reweight' to hook 2022-07-18 11:06:44 +08:00
renqin
99de0c1aeb [Refactor]: refactor RelativeLoc model and RandomPatchWithLabels transform 2022-07-18 11:06:44 +08:00
renqin
e39e3bbac0 [Refactor]: refactor RotationPred model and RotationWithLabels transform 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
5dc3696df7 [Refactor] refactor DeepCluster and ODC hook and related files 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
d1b7466d9b [Refactor]: Refactor deepcluster 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
063288240a [Refactor] refactor hooks 2022-07-18 11:06:44 +08:00
renqin
9565da23a4 [Feature]: Add RandomRotation transform 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
2ed7e90052 [Feature] add RandomCrop 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
f19f8f4c31 [Refactor]: Delete stepfixcosine and create a sample for new lr scheduler 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
953f790328 [Refactor] refactor dataset base config 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
25a5f71c7e [Refactor]: Logging 2022-07-18 11:06:44 +08:00
renqin
9ad2f6e6d5 [Refactor]: refactor the interface of byol, densecl, swav, barlowtwins and simclr 2022-07-18 11:06:44 +08:00
fangyixiao18
c119c4677c add RandomResizedCrop 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
be1dd2f5c2 [Refactor]: Refactor CAE 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
e687aff595 [Refactor]: Change the interface of SimMIM 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
e87be11a98 [Refactor] refactor moco and mocov3 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
dfbe3f6235 [Feature] add ColorJitter 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
11c3a6a3af [Refactor] refactor SimSiam 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
c8c0526582 [Fix]: Fix UT and use LabelData 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
efb9255d7e [Fix]: Support numpy array in random crop resize with two pic 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
4dc2ff1b79 [Refactor] refactor train/test interface 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
2063a54c2a [Refactor]: Refactor the interface of MAE 2022-07-18 11:06:44 +08:00
renqin
3426b49b39 [Refactor]: refactor RandomRotation and RandomPatch 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
d9c7bd6a7b [Fix] revise __repr__ format of some transforms 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
80916a0966 [Feature]: Add base model 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
472a0ab548 [Refactor] revise default_rutime.py 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
fa9d148ee8 [Refactor] add MultiView transform wrapper 2022-07-18 11:06:44 +08:00