torch==1.10.0
torchvision==0.11.1
einops==0.3.2
pytorch-lightning==1.5.2