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

make adf compatible with 7 gpus

Browse Source
This commit is contained in:
James Betker 2022-06-14 21:49:26 -06:00
parent 6bc19d1328
commit 804b365d5f
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

@ -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)