| Metadata-Version: |
2.1 |
| Name: |
pytorch-benchmark |
| Version: |
0.3.6 |
| Summary: |
Easily benchmark PyTorch model FLOPs, latency, throughput, max allocated memory and energy consumption in one go. |
| Author: |
Lukas Hedegaard |
| Author-Email: |
lukasxhedegaard[at]gmail.com |
| Home-Page: |
https://github.com/LukasHedegaard/pytorch-benchmark |
| Keywords: |
deep learning,pytorch,AI,benchmark,speed,energy,memory |
| Classifier: |
Environment :: Console |
| Classifier: |
Natural Language :: English |
| Classifier: |
Development Status :: 3 - Alpha |
| Classifier: |
Intended Audience :: Developers |
| Classifier: |
Topic :: Scientific/Engineering :: Artificial Intelligence |
| Classifier: |
Topic :: Scientific/Engineering :: Image Recognition |
| Classifier: |
Topic :: Scientific/Engineering :: Information Analysis |
| Classifier: |
Topic :: Software Development :: Libraries |
| Classifier: |
Topic :: Software Development :: Testing |
| Classifier: |
Topic :: System :: Benchmark |
| Classifier: |
License :: OSI Approved :: Apache Software License |
| Classifier: |
Operating System :: OS Independent |
| Classifier: |
Programming Language :: Python :: 3 |
| Classifier: |
Programming Language :: Python :: 3.7 |
| Classifier: |
Programming Language :: Python :: 3.8 |
| Classifier: |
Programming Language :: Python :: 3.9 |
| Requires-Dist: |
torch (>=1.6) |
| Requires-Dist: |
ptflops (~=0.6.8) |
| Requires-Dist: |
tqdm (~=4.62) |
| Requires-Dist: |
numpy |
| Requires-Dist: |
psutil (>=5.9) |
| Requires-Dist: |
gputil (>=1.4) |
| Requires-Dist: |
py-cpuinfo (>=7.0) |
| Requires-Dist: |
pyyaml (>=6.0) |
| Requires-Dist: |
setuptools; extra == "build" |
| Requires-Dist: |
wheel; extra == "build" |
| Requires-Dist: |
twine; extra == "build" |
| Requires-Dist: |
black; extra == "dev" |
| Requires-Dist: |
flake8; extra == "dev" |
| Requires-Dist: |
flake8-black; extra == "dev" |
| Requires-Dist: |
isort (>=5.7); extra == "dev" |
| Requires-Dist: |
pytest (~=6.0); extra == "dev" |
| Requires-Dist: |
pytest-cov; extra == "dev" |
| Requires-Dist: |
ptflops (>=0.6); extra == "dev" |
| Requires-Dist: |
numpy; extra == "dev" |
| Requires-Dist: |
torchvision; extra == "dev" |
| Requires-Dist: |
sphinx (>=3.0); extra == "docs" |
| Requires-Dist: |
nbsphinx (>=0.8); extra == "docs" |
| Requires-Dist: |
pandoc (>=1.0); extra == "docs" |
| Requires-Dist: |
docutils (>=0.16); extra == "docs" |
| Requires-Dist: |
sphinxcontrib-fulltoc (>=1.0); extra == "docs" |
| Requires-Dist: |
sphinxcontrib-mockautodoc; extra == "docs" |
| Requires-Dist: |
sphinx-autodoc-typehints (>=1.0); extra == "docs" |
| Requires-Dist: |
sphinx-paramlinks (>=0.4.0); extra == "docs" |
| Requires-Dist: |
sphinx-togglebutton (>=0.2); extra == "docs" |
| Requires-Dist: |
sphinx-copybutton (>=0.3); extra == "docs" |
| Requires-Dist: |
sphinx-autoapi (>=1.7); extra == "docs" |
| Requires-Dist: |
m2r2 (>=0.2); extra == "docs" |
| Provides-Extra: |
build |
| Provides-Extra: |
dev |
| Provides-Extra: |
docs |
| Description-Content-Type: |
text/markdown |
| License-File: |
LICENSE |