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

And remove unused parameters

pull/2/head
James Betker 2022-07-19 15:05:12 +07:00
parent 17a07b2e33
commit b203a7dc97
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/trainer/eval/music_diffusion_fid.py
Unescape Escape View File

@ -151,9 +151,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
# x = x.clamp(-s, s) / s
# return x
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,
'conditioning_input': mel_norm,
'disable_diversity': True})
model_kwargs={'truth_mel': mel_norm})
gen_mel_denorm = denormalize_mel(gen_mel)
output_shape = (1,16,audio.shape[-1]//16)