- mmengine.analysis.activation_count(model, inputs, supported_ops=None)¶
Given a model and an input to the model, compute the total number of activations of the model.
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.
A dictionary that records the number of activation (mega) for each operation and a Counter that records the number of unsupported operations.
- Return type