mmengine.analysis.parameter_count

mmengine.analysis.parameter_count(model)

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:

dict[str, int]