Skip to content

moduleprofiler v0.0.1
Compare
Choose a tag to compare
@eagomez2 eagomez2 released this 09 Sep 21:26
· 32 commits to main since this release
v0.0.1
e2bc568

First release of moduleprofiler, a fee open-source package to profile torch.nn.Module modules and obtain useful information to design a model that fits your needs and constraints at development time.

With moduleprofiler you can:

  • Calculate the number of parameters of your model.
  • Trace the input and output sizes of each component of your model.
  • Estimate the number of operations your model performs in a forward pass.
  • Calculate per module and total inference time.

All results can be obtained in one of the following formats:

  • dict (default output format)
  • pandas.DataFrame (to perform further calculations or filtering in your code)
  • html (to export as webpage)
  • LaTeX (to include in your publications)

[Online documentation | Tutorial ]

Assets 2
Loading