pppppM
|
cb238e36e3
|
[Feature] Add Recorder to improve Distiller
|
2022-07-15 23:04:38 +08:00 |
|
PJDong
|
0939aecdb6
|
refactor name of mutable module
|
2022-07-15 23:04:38 +08:00 |
|
pppppM
|
577a3a2a94
|
[Refactor] Base Algorithm
|
2022-07-15 23:04:13 +08:00 |
|
pppppM
|
baa8c8614e
|
[Deprecated] Clean up code that will be deprecated in OpenMMLab 2.0
|
2022-07-15 23:02:37 +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 |
|
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 |
|
whcao
|
f7fbe770a6
|
add pytest (#18)
Co-authored-by: caoweihan <caoweihan@sensetime.com>
|
2021-12-23 11:29:12 +08:00 |
|