mmrazor

mirror of https://github.com/open-mmlab/mmrazor.git synced 2025-06-03 15:02:54 +08:00

History

* [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 )