mmrazor/tests/test_models/test_mutables
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_mutable_channel [Improvement] Update NasMutator to build search_space in NAS (#426) 2023-02-01 22:51:38 +08:00
__init__.py merge pruning into dev-1.x (#312) 2022-10-10 17:30:25 +08:00
test_derived_mutable.py [Feature] Add Autoformer algorithm (#315) 2022-11-14 13:01:04 +08:00
test_diffchoiceroute.py
test_diffop.py [Refactor] Refactor Mutables and Mutators (#324) 2022-11-01 12:49:42 +08:00
test_gumbelchoiceroute.py
test_mutable_value.py [Feature] Add Autoformer algorithm (#315) 2022-11-14 13:01:04 +08:00
test_onehotop.py [Refactor] Refactor Mutables and Mutators (#324) 2022-11-01 12:49:42 +08:00
test_oneshotop.py
test_sequential_mutable_channel.py [Feature] Add DCFF (#295) 2022-11-23 09:55:33 +08:00