Shortcuts

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

dict[str, int]

Read the Docs v: stable
Versions
latest
stable
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.