torch>=1.3.1
torchvision>=0.5.0
numpy>=1.9.1

[all]
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex
requests
tqdm
sklearn
matplotlib
torchinfo
pytest

[datasets]
requests
tqdm

[docs]
recommonmark
sphinx
sphinx_rtd_theme
sphinxcontrib-katex

[examples]
sklearn
matplotlib
torchinfo

[tests]
pytest
