mmengine.analysis

ActivationAnalyzer	Provides access to per-submodule model activation count obtained by tracing a model with pytorch's jit tracing functionality.
FlopAnalyzer	Provides access to per-submodule model flop count obtained by tracing a model with pytorch's jit tracing functionality.

activation_count	Given a model and an input to the model, compute the total number of activations of the model.
flop_count	Given a model and an input to the model, compute the per-operator Gflops of the given model.
parameter_count	Count parameters of a model and its submodules.
parameter_count_table	Format the parameter count of the model (and its submodules or parameters)
get_model_complexity_info	Interface to get the complexity of a model.