From 07f7be24ce91b181996a6f9f15c5ed18b59625c3 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 22 May 2022 20:00:55 -0600 Subject: [PATCH] :/ --- codes/trainer/eval/music_diffusion_fid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/trainer/eval/music_diffusion_fid.py b/codes/trainer/eval/music_diffusion_fid.py index 0d8723bc..80a0c580 100644 --- a/codes/trainer/eval/music_diffusion_fid.py +++ b/codes/trainer/eval/music_diffusion_fid.py @@ -181,7 +181,7 @@ class MusicDiffusionFid(evaluator.Evaluator): #gen_wav = pixel_shuffle_1d(gen_wav, 16) #return gen_wav, real_resampled, gen_mel, mel_norm, sample_rate - return real_resampled, real_resampled, gen_mel, mel100_norm, sample_rate + return real_resampled.unsqueeze(0), real_resampled, gen_mel, mel100_norm, sample_rate def project(self, sample, sample_rate): sample = torchaudio.functional.resample(sample, sample_rate, 22050)