einops~=0.3.0
numpy~=1.19.5
torchvision~=0.10.0
antialiased_cnns
axial_attention
pytorch_msssim
torch