torch>=1.3.1
torchvision>=0.7.3
numpy>=1.9.1
sklearn
matplotlib

[all]
requests
tqdm
bump2version
pre-commit
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex
sphinx-autodoc-typehints
torchinfo
flake8
pytest

[datasets]
requests
tqdm

[dev]
bump2version
pre-commit

[docs]
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex
sphinx-autodoc-typehints

[examples]
torchinfo

[tests]
flake8
pytest
