Commit Graph

619 Commits (68129f0a9c84d89057429116021d0544091bb12a)
 

Author SHA1 Message Date
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
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