From f31a333c4ff51eef9d5d4d1091fce7a610159b80 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 10 Oct 2022 20:11:28 -0600 Subject: [PATCH] more sampling fixes --- 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 2e157bdd..ca2a5202 100644 --- a/codes/trainer/eval/audio_diffusion_fid.py +++ b/codes/trainer/eval/audio_diffusion_fid.py @@ -191,7 +191,7 @@ class AudioDiffusionFid(evaluator.Evaluator): gen_wav = self.local_modules['vocoder'].inference(gen_mel) real_dec = self.local_modules['vocoder'].inference(univnet_mel) - return gen_wav.float(), real_dec, SAMPLE_RATE + return gen_wav.float(), real_dec, gen_mel, univnet_mel, SAMPLE_RATE def perform_diffusion_ctc(self, audio, codes, text): SAMPLE_RATE = 24000