gitpython
h5py
kornia
lz4
matplotlib
numpy
opencv-python
orion
pillow>=7.1
pyyaml
scikit-learn
six
tensorboardX
torch
torchvision>=0.6
tqdm

[rst]
docutils>=0.11
