revsic / jax-variational-diffwave Goto Github PK

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

Hi, I have been trying to implement VDM as well, and am just wondering whether you have had any success with this?

I am working in PyTorch with the discrete-time version, but I am having trouble with numerical stability. It is not directly comparable, but most of our code seems to match up.

One difference I notice is that you constrained the biases of logSNR function to be positive - I think you can get away with positive weights and unconstrained biases (to satisfy monotonic constraint), but I'm willing to be corrected here. (Not sure if this makes much of a difference).

Otherwise, this is a really awesome implementation! I'm just curious if it works because I am struggling to make my own version work!

Cheers