jax-variational-diffwave icon indicating copy to clipboard operation
jax-variational-diffwave copied to clipboard

Evaluation code

Open matrix-alpha opened this issue 3 years ago • 1 comments

Hi, thanks for implementation. Can you provide the evaluation code?

matrix-alpha avatar Jun 23 '22 22:06 matrix-alpha

Sample inference code is here, or did you mean KDSD, FDSD-like evaluations ?

with open('l1.json') as f:
    config = Config.load(json.load(f))

diffwave = VLBDiffWaveApp(config.model)
diffwave.restore('./l1/l1_99.ckpt')

# mel: [B, T, mel]
audio, _ = diffwave(mel, timesteps=50, key=jax.random.PRNGKey(0))

revsic avatar Jun 26 '22 12:06 revsic