deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

:/

Browse Source
This commit is contained in:
James Betker 2022-05-22 20:00:55 -06:00
parent c0bc466aad
commit 07f7be24ce
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/music_diffusion_fid.py
View File

@ -181,7 +181,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
#gen_wav = pixel_shuffle_1d(gen_wav, 16) #gen_wav = pixel_shuffle_1d(gen_wav, 16)
#return gen_wav, real_resampled, gen_mel, mel_norm, sample_rate #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): def project(self, sample, sample_rate):
sample = torchaudio.functional.resample(sample, sample_rate, 22050) sample = torchaudio.functional.resample(sample, sample_rate, 22050)