torch>=1.9.0
torchvision>=0.10.0

[dev]
black>=21.5b
flake8>=3.9
isort>=5.9
