mirrors
/
mmrazor
mirror of https://github.com/open-mmlab/mmrazor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmrazor/tests/test_models
History
Yang Gao 8d603d917e
[Feature] Add Dsnas Algorithm (#226) 
* [tmp] Update Dsnas

* [tmp] refactor arch_loss & flops_loss

* Update Dsnas & MMRAZOR_EVALUATOR:
1. finalized compute_loss & handle_grads in algorithm;
2. add MMRAZOR_EVALUATOR;
3. fix bugs.

* Update lr scheduler & fix a bug:
1. update param_scheduler & lr_scheduler for dsnas;
2. fix a bug of switching to finetune stage.

* remove old evaluators

* remove old evaluators

* update param_scheduler config

* merge dev-1.x into gy/estimator

* add flops_loss in Dsnas using ResourcesEstimator

* get resources before mutator.prepare_from_supernet

* delete unness broadcast api from gml

* broadcast spec_modules_resources when estimating

* update early fix mechanism for Dsnas

* fix merge

* update units in estimator

* minor change

* fix data_preprocessor api

* add flops_loss_coef

* remove DsnasOptimWrapper

* fix bn eps and data_preprocessor

* fix bn weight decay bug

* add betas for mutator optimizer

* set diff_rank_seed=True for dsnas

* fix start_factor of lr when warm up

* remove .module in non-ddp mode

* add GlobalAveragePoolingWithDropout

* add UT for dsnas

* remove unness channel adjustment for shufflenetv2

* update supernet configs

* delete unness dropout

* delete unness part with minor change on dsnas

* minor change on the flag of search stage

* update README and subnet configs

* add UT for OneHotMutableOP
 		2022-09-29 16:48:47 +08:00
..
test_algorithms [Feature] Add Dsnas Algorithm (#226) 2022-09-29 16:48:47 +08:00
test_architectures [feature] CONTRASTIVE REPRESENTATION DISTILLATION with dataset wrapper (#281) 2022-09-13 20:53:43 +08:00
test_distillers [fix] fix mmcv mmengine (#242) 2022-08-24 09:58:11 +08:00
test_losses [feature] CONTRASTIVE REPRESENTATION DISTILLATION with dataset wrapper (#281) 2022-09-13 20:53:43 +08:00
test_mutables [Feature] Add Dsnas Algorithm (#226) 2022-09-29 16:48:47 +08:00
test_mutators [Feature] Add Zero-shot Knowledge Transfer via Adversarial Belief Matching (#241) 2022-08-24 14:57:35 +08:00
test_subnet [Feature] Refactor Estimator for computing FLOPs/Params/Latency. (#230) 2022-08-23 15:01:47 +08:00
test_task_modules/test_estimators [Improvement] Update estimator with api revision (#277) 2022-09-14 20:39:49 +08:00