torch==1.7.0
torchvision==0.8.1
einops==0.3.0