torch>=1.3.1
torchvision<0.5.0,>=0.4.0
yacs==0.1.6
pytorch-lightning>=0.6.0
pretrainedmodels>=0.7.4
pillow<7.0.0,>=6.2.0
test-tube==0.7.5
albumentations>=0.4.3
