And remove unused parameters
This commit is contained in:
parent
17a07b2e33
commit
b203a7dc97
|
@ -151,9 +151,7 @@ class MusicDiffusionFid(evaluator.Evaluator):
|
||||||
# x = x.clamp(-s, s) / s
|
# x = x.clamp(-s, s) / s
|
||||||
# return x
|
# return x
|
||||||
gen_mel = self.diffuser.p_sample_loop(self.model, mel_norm.shape, #denoised_fn=denoising_fn, clip_denoised=False,
|
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,
|
model_kwargs={'truth_mel': mel_norm})
|
||||||
'conditioning_input': mel_norm,
|
|
||||||
'disable_diversity': True})
|
|
||||||
|
|
||||||
gen_mel_denorm = denormalize_mel(gen_mel)
|
gen_mel_denorm = denormalize_mel(gen_mel)
|
||||||
output_shape = (1,16,audio.shape[-1]//16)
|
output_shape = (1,16,audio.shape[-1]//16)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user