mmrazor/tests/test_models/test_architectures/test_backbones
Yang Gao a27952dbb1
[Improvement] Update NasMutator to build search_space in NAS (#426)
* update space_mixin

* update NAS algorithms with SpaceMixin

* update pruning algorithms with SpaceMixin

* fix ut

* fix comments

* revert _load_fix_subnet_by_mutator

* fix dcff test

* add ut for registry

* update autoslim_greedy_search

* fix repeat-mutables bug

* fix slice_weight in export_fix_subnet

* Update NasMutator:
1. unify mutators for NAS algorithms as the NasMutator;
2. regard ChannelMutator as pruning-specified;
3. remove value_mutators & module_mutators;
4. set GroupMixin only for NAS;
5. revert all changes in ChannelMutator.

* update NAS algorithms using NasMutator

* update channel mutator

* update one_shot_channel_mutator

* fix comments

* update UT for NasMutator

* fix isort version

* fix comments

---------

Co-authored-by: gaoyang07 <1546308416@qq.com>
Co-authored-by: liukai <your_email@abc.example>
2023-02-01 22:51:38 +08:00
..
test_autoformerbackbone.py [Feature] Add Autoformer algorithm (#315) 2022-11-14 13:01:04 +08:00
test_dartsbackbone.py [Improvement] Update NasMutator to build search_space in NAS (#426) 2023-02-01 22:51:38 +08:00
test_searchable_mobilenet_v2.py [Feature] Add BigNAS algorithm (#219) 2022-12-07 11:28:10 +08:00
test_searchable_mobilenet_v3.py [Improvement] Adapt OFA series with SearchableMobileNetV3 (#385) 2022-12-15 22:19:55 +08:00
test_searchable_shufflenet_v2.py Refactor subnet sampler 2022-07-15 23:04:38 +08:00
utils.py Refactor backbone 2022-07-15 23:04:13 +08:00