Shortcuts

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.

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