mmengine.analysis.parameter_count_table¶
- mmengine.analysis.parameter_count_table(model, max_depth=3)[源代码]¶
Format the parameter count of the model (and its submodules or parameters)
Adopted from https://github.com/facebookresearch/fvcore/blob/main/fvcore/nn/parameter_count.py