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

put frechet_distance on cuda

Browse Source
This commit is contained in:
James Betker 2022-02-23 21:21:13 -07:00
parent 9a7bbf33df
commit 81017d9696
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -140,7 +140,7 @@ class AudioDiffusionFid(evaluator.Evaluator):
torchaudio.save(os.path.join(save_path, f"{self.env['rank']}_{i}_real.wav"), ref.squeeze(0).cpu(), sample_rate)
gen_projections = torch.stack(gen_projections, dim=0)
real_projections = torch.stack(real_projections, dim=0)
frechet_distance = self.compute_frechet_distance(gen_projections, real_projections)
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()