torch==1.8.1
torchvision==0.9.1
einops==0.3.0
pytorch-lightning==1.2.8