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