mmrazor/tests/test_models/test_mutators
qiufeng d190037a5e
[Feature] Add dynamic bricks (#228)
* add dynamic bricks

* add dynamic conv2d test

* add tests for dynamic linear and dynamic norm

* add docstring for dynamic conv2d

* add docstring for dynamic linear

* add docstring for dynamic batchnorm

* Refactor the dynamic op ( put more logic into the mixin )

* fix UT

* Fix UT ( fileio was moved to mmengine)

* derived mutable adds choices property

* Unify the register interface of mutable in dynamic op

* Unified getter interface of mutable in dynamic op

Co-authored-by: gaojianfei <gaojianfei@sensetime.com>
Co-authored-by: pppppM <gjf_mail@126.com>
2022-08-19 15:15:38 +08:00
..
test_classical_models [Refactor] Move `build_arch_param` from `DiffMutableModule` to `DiffModuleMutator` (#221) 2022-08-10 10:05:32 +08:00
__init__.py [Feature] Add slimmable algorithm 2022-07-15 23:05:07 +08:00
test_channel_mutator.py [Feature] Add slimmable algorithm 2022-07-15 23:05:07 +08:00
test_diff_mutator.py [Refactor] Move `build_arch_param` from `DiffMutableModule` to `DiffModuleMutator` (#221) 2022-08-10 10:05:32 +08:00
test_one_shot_mutator.py Align SPOS and DetNAS to MMRazor2.0 2022-07-15 23:04:38 +08:00
utils.py [Feature] Add dynamic bricks (#228) 2022-08-19 15:15:38 +08:00