PyPI page
Home page
Author:
Lukas Hedegaard
License:
Summary:
Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption in one go.
Latest version:
0.3.6
Required dependencies:
gputil
|
numpy
|
psutil
|
ptflops
|
py-cpuinfo
|
pyyaml
|
torch
|
tqdm
Optional dependencies:
black
|
docutils
|
flake8
|
flake8-black
|
isort
|
m2r2
|
nbsphinx
|
numpy
|
pandoc
|
ptflops
|
pytest
|
pytest-cov
|
setuptools
|
sphinx
|
sphinx-autoapi
|
sphinx-autodoc-typehints
|
sphinx-copybutton
|
sphinx-paramlinks
|
sphinx-togglebutton
|
sphinxcontrib-fulltoc
|
sphinxcontrib-mockautodoc
|
torchvision
|
twine
|
wheel
Downloads last day:
17
Downloads last week:
449
Downloads last month:
967