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

fix mdf

Browse Source
This commit is contained in:
James Betker 2022-07-19 19:39:29 -06:00
parent c14bf6dfb2
commit 2997a640b0
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

@ -153,7 +153,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
gen_mel = self.diffuser.p_sample_loop(self.model, mel_norm.shape, #denoised_fn=denoising_fn, clip_denoised=False,
model_kwargs={'truth_mel': mel_norm})
gen_mel_denorm = denormalize_mel(gen_mel)
gen_mel_denorm = denormalize_torch_mel(gen_mel)
output_shape = (1,16,audio.shape[-1]//16)
self.spec_decoder = self.spec_decoder.to(audio.device)
gen_wav = self.spectral_diffuser.p_sample_loop(self.spec_decoder, output_shape,