PyYAML>=5.3.1
requests
GitPython>=3.1.24
torch>=1.7.0
torchvision>=0.8.1
psutil
IPython

[tests]
pytest
pytest-cov
coverage
