mmrazor/tests
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
..
data [feature] CONTRASTIVE REPRESENTATION DISTILLATION with dataset wrapper (#281) 2022-09-13 20:53:43 +08:00
test_core [Fix] Fix tracer (#273) 2022-09-15 14:52:54 +08:00
test_datasets [feature] CONTRASTIVE REPRESENTATION DISTILLATION with dataset wrapper (#281) 2022-09-13 20:53:43 +08:00
test_models [Feature] Add Dsnas Algorithm (#226) 2022-09-29 16:48:47 +08:00
test_registry Reorganize Dirs (#205) 2022-07-25 13:28:08 +08:00
test_runners [Improvement] Update estimator with api revision (#277) 2022-09-14 20:39:49 +08:00
__init__.py add BaseNode/BaseGraph, ModuleNode/ModuleGraph (#217) 2022-08-19 15:38:19 +08:00
test_metafiles.py Add benchmark tools & Reorgnazie configs 2022-07-15 23:05:10 +08:00