torch==1.7.1
torchvision==0.8.2
pytorch_lightning==1.1.0
scipy==1.5.4
