torch~=1.3
torchvision~=0.5

[dev]
black
flake8
flake8-blind-except
flake8-bugbear
flake8-builtins
flake8-docstrings
flake8-import-order
flake8-mutable
mypy
pre-commit
pytest
pytest-cov
pytest-xdist

[tensorboard]
tensorflow>=1.0
