Shortcuts

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

参数:

model (nn.Module) – the model to count parameters.

返回:

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.

返回类型:

dict[str, int]

Read the Docs v: latest
Versions
latest
stable
v0.10.3
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.