Shortcuts

mmengine.analysis.activation_count

mmengine.analysis.activation_count(model, inputs, supported_ops=None)[source]

Given a model and an input to the model, compute the total number of activations of the model.

Adopted from https://github.com/facebookresearch/fvcore/blob/main/fvcore/nn/activation_count.py

Parameters:
  • model (nn.Module) – The model to compute activation counts.

  • inputs (tuple) – Inputs that are passed to model to count activations. Inputs need to be in a tuple.

  • supported_ops (dict(str,Callable) or None) – provide additional handlers for extra ops, or overwrite the existing handlers for convolution and matmul. The key is operator name and the value is a function that takes (inputs, outputs) of the op.

Returns:

A dictionary that records the number of activation (mega) for each operation and a Counter that records the number of unsupported operations.

Return type:

tuple[defaultdict, Counter]

Read the Docs v: v0.10.4
Versions
latest
stable
v0.10.4
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
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.