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]