torch>=1.3
torchvision>=0.4.2
yacs==0.1.6
pytorch-lightning==0.5.3.2
pretrainedmodels
