fix an issue with new version of pytorch fid

Author: lucidrains

stylegan2_pytorch/stylegan2_pytorch.py

@@ -1145,7 +1145,7 @@ def calculate_fid(self, num_batches):
             for j in range(generated_images.size(0)):
                 torchvision.utils.save_image(generated_images[j, :, :, :], str(Path(fake_path) / f'{str(j + batch_num * self.batch_size)}-ema.{ext}'))
 
-        return fid_score.calculate_fid_given_paths([real_path, fake_path], 256, True, 2048)
+        return fid_score.calculate_fid_given_paths([real_path, fake_path], 256, latents.device, 2048)
 
     @torch.no_grad()
     def truncate_style(self, tensor, trunc_psi = 0.75):

stylegan2_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.5.11'
+__version__ = '1.5.12'