mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gurgl

Browse Source
This commit is contained in:
James Betker 2022-02-23 21:28:24 -07:00
parent e6824e398f
commit 7c17c8e674
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/trainer/eval/audio_diffusion_fid.py
View File

@ -143,7 +143,8 @@ class AudioDiffusionFid(evaluator.Evaluator):
frechet_distance = torch.tensor(self.compute_frechet_distance(gen_projections, real_projections), device=self.env['device'])
if distributed.is_initialized() and distributed.get_world_size() > 1:
frechet_distance = distributed.all_reduce(frechet_distance) / distributed.get_world_size()
distributed.all_reduce(frechet_distance)
frechet_distance = frechet_distance / distributed.get_world_size()
self.model.train()
if hasattr(self, 'dvae'):