only calculate fid on main process

Author: lucidrains

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

@@ -912,8 +912,9 @@ def __init__(
 
         # FID-score computation
 
-        self.calculate_fid = calculate_fid
-        if calculate_fid:
+        self.calculate_fid = calculate_fid and self.accelerator.is_main_process
+
+        if self.calculate_fid:
             if not self.model.is_ddim_sampling:
                 self.accelerator.print(
                     "WARNING: Robust FID computation requires a lot of generated samples and can therefore be very time consuming."\

denoising_diffusion_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.8.1'
+__version__ = '1.8.2'