torch==1.8.1
torchvision==0.9.1
pytorch_lightning==1.2.6
scipy==1.5.4

