fix SSIM error

evaluation/helpers/metrics.py

@@ -14,8 +14,8 @@ from evaluation.helpers.score_infinity import calculate_FID_infinity_path, calcu
 
 def image_to_tensor(path, sample='all', device='cpu'):
 
-    transform_resize = transforms.Compose([transforms.ToTensor(), transforms.Resize(128), transforms.Lambda(lambda x: (x * 255).type(torch.uint8))])
-    transform = transforms.Compose([transforms.ToTensor(), transforms.Lambda(lambda x: (x * 255).type(torch.uint8)) ])
+    transform_resize = transforms.Compose([transforms.ToTensor(), transforms.Resize(128), transforms.Lambda(lambda x: (x * 255))])
+    transform = transforms.Compose([transforms.ToTensor(), transforms.Lambda(lambda x: (x * 255)) ])
     filelist = []
     for p, subdirs, files in os.walk(path):
         for f in files: