torch>=1.8.1
einops>=0.3.0
torchvision>=0.8.2
