de4dd13cc6
[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>
2022-04-02 19:16:41 +08:00
45b8ce7448
reset norm running status after prepare_from_supernet ( #81 )
2022-04-02 14:06:56 +08:00
0ab5efeb9e
[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>
2022-04-01 19:06:20 +08:00
20d1e0b0d8
[Refactor] Delete redundant `set_random_seed` function ( #104 )
...
* refactor set_random_seed
* add unittests
* fix unittests error
* fix lint
* avoid bc breaking
2022-03-16 22:08:26 +08:00
5d925df56a
Merge branch 'master' of https://github.com/open-mmlab/mmrazor into dev_v0.3.0
2022-03-08 15:23:01 +08:00
f48509aebd
[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
2022-03-08 11:38:40 +08:00
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
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
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
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
f7fbe770a6
add pytest ( #18 )
...
Co-authored-by: caoweihan <caoweihan@sensetime.com>
2021-12-23 11:29:12 +08:00
whcao
qiufeng
wutongshenqiu
pppppM