einops
torchvision==0.9.0
torch==1.8.0
scipy==1.0.0
tqdm