mmrazor/tests/test_models
Yang Gao 4b3f8ab69e
[Feature] Refactor Estimator for computing FLOPs/Params/Latency. (#230)
* Refactor ModelEstimator:
1. add EvaluatorLoop in engine.runners;
2. add estimator for structures (both subnet & supernet);
3. add layer_counter for each op.

* fix lint

* update estimator:
1. add ResourceEstimator based on BaseEstimator;
2. add notes & examples for ResourceEstimator & EvaluatorLoop usage;
3. fix a bug of latency test.
4. minor changes according to comments.

* add UT & fix a bug caused by UT

* add docstrings & remove old estimator

* update docstrings for op_spec_counters

* rename resource_evaluator_val_loop

* support adding resource attrs of each submodule in a measured model

* fix lint

* refactor estimator file structures

* support estimating resources for spec modules

* rm old UT

* update new estimator UT cases

* fix traversal range of the model

* cancel unit convert in accumulate_sub_module_flops_params

* use estimator_cfg to build ResourceEstimator

* fix a broadcast bug

* delete fixed input_shape

* add assertion and string-format-return when measuring spec_modules

* add UT for estimating spec_modules
2022-08-23 15:01:47 +08:00
..
test_algorithms [Feature] Add DAFL Distillation (#235) 2022-08-23 10:47:34 +08:00
test_architectures [Feature] Add DAFL Distillation (#235) 2022-08-23 10:47:34 +08:00
test_distillers [Feature] Add BYOT Distillation (#232) 2022-08-22 14:08:02 +08:00
test_losses [Feature] Add DAFL Distillation (#235) 2022-08-23 10:47:34 +08:00
test_mutables [Feature] Add `DerivedMutable` & `MutableValue` (#215) 2022-08-10 16:07:03 +08:00
test_mutators [Feature] Add dynamic bricks (#228) 2022-08-19 15:15:38 +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 [Feature] Refactor Estimator for computing FLOPs/Params/Latency. (#230) 2022-08-23 15:01:47 +08:00