- 19 Mar, 2021 1 commit
-
-
zhouzaida authored
* [Feature] support to calculate FLOPs of GN, IN, LN * Update flops_counter.py * Update flops_counter.py
-
- 26 Jun, 2020 1 commit
-
-
Jintao Lin authored
* Add utils to calculate model complexity info * remove _InstanceNorm in unittest * add docstring and increase unittest coverage * fix deconv_flops_counter_hook to accept different data shape * test when model is not a common instance * put flops_counter.py and weight_init.py into mmcv/cnn/utils folder * fix import name * reformat some docstrings * update the scripts with latest one and remove redundant codes * directly represent a model without string and eval() * reformat code
-