from MS_SSIM_L1_loss import MS_SSIM_L1_LOSS criterion = MS_SSIM_L1_LOSS() # your pytorch tensor x, y with [B, C, H, W] dimension on cuda device 0 loss = criterion(x, y) Please check demo.py for more details. Requirements: ...