torch>=1.6
ptflops~=0.6.8
tqdm~=4.62
numpy
psutil>=5.9
gputil>=1.4
py-cpuinfo>=7.0
pyyaml>=6.0
carbontracker>=1.1

[build]
setuptools
wheel
twine

[dev]
black
flake8
flake8-black
isort>=5.7
pytest~=6.0
pytest-cov
ptflops>=0.6
numpy
torchvision

[docs]
sphinx>=3.0
nbsphinx>=0.8
pandoc>=1.0
docutils>=0.16
sphinxcontrib-fulltoc>=1.0
sphinxcontrib-mockautodoc
sphinx-autodoc-typehints>=1.0
sphinx-paramlinks>=0.4.0
sphinx-togglebutton>=0.2
sphinx-copybutton>=0.3
sphinx-autoapi>=1.7
m2r2>=0.2
