mmrazor/tests
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
..
data add BaseNode/BaseGraph, ModuleNode/ModuleGraph (#217) 2022-08-19 15:38:19 +08:00
test_core add BaseNode/BaseGraph, ModuleNode/ModuleGraph (#217) 2022-08-19 15:38:19 +08:00
test_models [Feature] Refactor Estimator for computing FLOPs/Params/Latency. (#230) 2022-08-23 15:01:47 +08:00
test_registry Reorganize Dirs (#205) 2022-07-25 13:28:08 +08:00
test_runners [Feature] Refactor Estimator for computing FLOPs/Params/Latency. (#230) 2022-08-23 15:01:47 +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