mmengine.analysis.parameter_count¶
- mmengine.analysis.parameter_count(model)[source]¶
Count parameters of a model and its submodules.
Adopted from https://github.com/facebookresearch/fvcore/blob/main/fvcore/nn/parameter_count.py
- Parameters
model (nn.Module) – the model to count parameters.
- Returns
the key is either a parameter name or a module name. The value is the number of elements in the parameter, or in all parameters of the module. The key “” corresponds to the total number of parameters of the model.
- Return type