diff --git a/evaluation/helpers/metrics.py b/evaluation/helpers/metrics.py
index 93b11a4ad0236930e5a8ec852c0da2866cfb9c78..5fcb4be7cab3ccaaae5f025cc7d35d00fe463541 100644
--- a/evaluation/helpers/metrics.py
+++ b/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: