humu789
8913d6840d
[Feature] Add KD val loop
2022-07-15 23:04:38 +08:00
whcao
6bdd9d682b
[Refactor] Rename ChannelMutable to MutableChannel
2022-07-15 23:04:38 +08:00
PJDong
0939aecdb6
refactor name of mutable module
2022-07-15 23:04:38 +08:00
whcao
42063ae4d3
[Refactor]Refactor tracer and channel mutator
2022-07-15 23:04:35 +08:00
PJDong
332f49ac6f
Support SubnetMixin and add Razor Registry Build Function
2022-07-15 23:04:13 +08:00
qiufeng
2d17ac6c2e
[Feature] Add flops estimator
2022-07-15 23:04:13 +08:00
whcao
b1ac66a9a3
[Features]Weihan dev/delivery
2022-07-15 23:04:13 +08:00
PJDong
56afc69d85
Refactor DartsBackbone
2022-07-15 23:04:13 +08:00
pppppM
577a3a2a94
[Refactor] Base Algorithm
2022-07-15 23:04:13 +08:00
pppppM
21e2110673
Refactor Diff Mutator
2022-07-15 23:04:13 +08:00
PJDong
804a595fc6
Refactor DiffMutable, DiffOP and DiffChoiceRoute.
2022-07-15 23:04:13 +08:00
qiufeng
99e7993376
Refactor backbone
2022-07-15 23:04:13 +08:00
qiufeng
910b131183
Refactor Mutator
2022-07-15 23:04:13 +08:00
PJDong
3e4b04a02f
Refactor Mutable
2022-07-15 23:04:13 +08:00
pppppM
4b5ddaf642
[Refactor] Refactor Registry
2022-07-15 23:02:57 +08:00
pppppM
2efe63e9a4
[CI] Add Gitlab CI
2022-07-15 23:02:57 +08:00
pppppM
baa8c8614e
[Deprecated] Clean up code that will be deprecated in OpenMMLab 2.0
2022-07-15 23:02:37 +08:00
qiufeng
16c643d946
[Fix] Fix init mmdet model api ( #179 )
...
* fix init_mmdet_model
* add test
2022-06-15 19:21:01 +08:00
pppppM
2dad24044d
Bump version to 0.3.1 ( #155 )
...
* [Enhance] Add extra dataloader settings in configs (#141 )
* [Docs] fix md link failure in docs (#142 )
* [Docs] update Cream readme
* delete 'readme.md' in model_zoo.md
* fix md link failure in docs
* [Docs] add myst_parser to extensions in conf.py
* [Docs] delete the deprecated recommonmark
* [Docs] delete recommandmark from conf.py
* [Docs] fix md link failure and lint failture
* [Fix] Fix seed error in mmseg/train_seg.py and typos in train.md (#152 )
* [Docs] update Cream readme
* delete 'readme.md' in model_zoo.md
* fix cwd docs and fix seed in #151
* delete readme of cream
* [Enhancement]Support broadcast_object_list in multi-machines & support Searcher running in single GPU (#153 )
* broadcast_object_list support multi-machines
* add userwarning
* [Fix] Fix configs (#149 )
* fix configs
* fix spos configs
* fix readme
* replace the official mutable_cfg with the mutable_cfg searched by ourselves
* update https prefix
Co-authored-by: pppppM <gjf_mail@126.com>
* [BUG]Support to prune models containing GroupNorm or InstanceNorm. (#144 )
* suport GN and IN
* test pruner
* limit pytorch version
* fix pytest
* throw an error when tracing groupnorm with torch version under 1.6.0
Co-authored-by: caoweihan <caoweihan@sensetime.com>
* Bump version to 0.3.1
Co-authored-by: qiufeng <44188071+wutongshenqiu@users.noreply.github.com>
Co-authored-by: PJDong <1115957667@qq.com>
Co-authored-by: humu789 <88702197+humu789@users.noreply.github.com>
Co-authored-by: whcao <41630003+HIT-cwh@users.noreply.github.com>
Co-authored-by: caoweihan <caoweihan@sensetime.com>
2022-05-05 01:02:45 +08:00
pppppM
49f1bee45b
Bump version to v0.3.0 ( #135 )
...
* [Feature] Add function to meet mmdeploy support (#102 )
* add init_model function for mmdeploy
* fix lint
* add unittest for init_xxx_model
* fix lint
* mv test_inference.py to test_apis directory
* [Feature] Add function to meet mmdeploy support (#102 )
* add init_model function for mmdeploy
* fix lint
* add unittest for init_xxx_model
* fix lint
* mv test_inference.py to test_apis directory
* [Refactor] Delete redundant `set_random_seed` function (#104 )
* refactor set_random_seed
* add unittests
* fix unittests error
* fix lint
* avoid bc breaking
* [Feature] Add diff seeds to diff ranks and set torch seed in worker_init_fn (#113 )
* add init_random_seed
* Set diff seed to diff workers
* [Feature] Add multi machine dist_train (#114 )
* support multi nodes
* update training doc
* fix lints
* remove fixed seed
* fix ddp wrapper registry (#128 )
* [Docs] Add brief installation steps in README(_zh-CN).md (#121 )
* Add brief installation
* add brief installtion ref to mmediting pr#816
Co-authored-by: caoweihan <caoweihan@sensetime.com>
* [BUG]Fix bugs in pruner (#126 )
* fix bugs in pruner when pruning models with shared modules
* pruner can trace models with dilation conv2d
* fix deploy_subnet
* fix add_pruning_attrs
* fix bugs in modify_forward
* fix lint
* fix StructurePruner
* test tracing models with shared modules
Co-authored-by: caoweihan <caoweihan@sensetime.com>
* [Docs]Add some more details to docs (#133 )
* add docs for dataset
* add cfg-options for distillation
* fix docs
Co-authored-by: caoweihan <caoweihan@sensetime.com>
* reset norm running status after prepare_from_supernet (#81 )
* [Improvement]Sync train api (#115 )
Co-authored-by: caoweihan <caoweihan@sensetime.com>
* [Feature]Support Relational Knowledge Distillation (#127 )
* add rkd
* add rkd pytest
* add rkd configs
* fix readme
* fix rkd
* split rkd loss to distance-wise and angle-wise losses
* rename rkd losses
* add rkd metaflie
* add rkd related links
* rename rkd metafile and add to model index
* delete cifar100
Co-authored-by: caoweihan <caoweihan@sensetime.com>
Co-authored-by: pppppM <gjf_mail@126.com>
Co-authored-by: qiufeng <44188071+wutongshenqiu@users.noreply.github.com>
Co-authored-by: wutongshenqiu <690364065@qq.com>
Co-authored-by: whcao <41630003+HIT-cwh@users.noreply.github.com>
Co-authored-by: caoweihan <caoweihan@sensetime.com>
2022-04-02 19:30:50 +08:00
whcao
81e0e3452a
[Feature] Resume from the latest checkpoint automatically. ( #61 )
...
* support auto-resume
* support auto-resume
* support auto-resume
* support auto-resume
Co-authored-by: pppppM <67539920+pppppM@users.noreply.github.com>
2022-03-08 11:25:19 +08:00
pppppM
ef1637e866
[FEATURE] Mobilenet Series Search Space ( #82 )
...
* add mbv2 block and identity
* add mbv2 block and identity unittests
* expand_factor -> expand_ratio
* add searchable mobilenet v2
* fix lints
* add spos subnet retraining config
* fix mmcls slurm search
* add proxyless_gpu setting for mbv2
* use bn default
* add angelnas spos config
* update spos readme
* fix SELayer's useage
* add docstring
* rename mbv2 to mb
* add some unittest of mb
* rename mb to mobilenet
* add some rename-mb in configs
* update README of spos
* add rename-mb in unittest
* update test_mmcls
Co-authored-by: wutongshenqiu <690364065@qq.com>
Co-authored-by: humu789 <humu@pjlab.org.cn>
2022-03-07 23:15:18 +08:00
pppppM
c5a097b150
Add a new pre-commit-hook to automatically add a copyright ( #96 )
...
* Add a new pre-commit-hook to automatically add a copyright
* add check-algo-readme
Co-authored-by: qiufeng <44188071+wutongshenqiu@users.noreply.github.com>
* fix alg-readme lints
Co-authored-by: qiufeng <44188071+wutongshenqiu@users.noreply.github.com>
2022-03-02 20:29:28 +08:00
qiufeng
64ccbc03fd
[Enhance] Add setup multi-processes for all tasks ( #59 )
...
* Add setup-multi-processes for all tasks
* Add setup-multi-processes for all tasks
* Add test for setup-multi-processes
2022-01-26 19:17:55 +08:00
whcao
f7fbe770a6
add pytest ( #18 )
...
Co-authored-by: caoweihan <caoweihan@sensetime.com>
2021-12-23 11:29:12 +08:00