torch>=1.8
pytorch-msssim
