pytorch-lightning>=0.8.4
torch>=1.4
torchvision>=0.5
scikit-learn>=0.23
opencv-python
test_tube>=0.7.5
trains>=0.14.1
