From 804b365d5fb8747b4d9f2cb2429c7d633a17c07c Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 14 Jun 2022 21:49:26 -0600 Subject: [PATCH] make adf compatible with 7 gpus --- codes/trainer/eval/audio_diffusion_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/eval/audio_diffusion_fid.py b/codes/trainer/eval/audio_diffusion_fid.py index 742c58d5..91d6eab7 100644 --- a/codes/trainer/eval/audio_diffusion_fid.py +++ b/codes/trainer/eval/audio_diffusion_fid.py @@ -286,7 +286,7 @@ class AudioDiffusionFid(evaluator.Evaluator): real_projections = [] intelligibility_losses = [] for i in tqdm(list(range(0, len(self.data), self.skip))): - path, text, codes = self.data[i + self.env['rank']] + path, text, codes = self.data[(i + self.env['rank']) % len(self.data)] audio = load_audio(path, 22050).to(self.dev) codes = codes.to(self.dev) sample, ref, gen_mel, ref_mel, sample_rate = self.diffusion_fn(audio, codes, text)