torch==1.7.1
torchvision==0.8.2
einops==0.3.0
pytorch-lightning==1.1.4