search

rtqichen / torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
MIT License
5.61k stars 930 forks source link

[FFJORD] How to sample from vae flow? #215

Open bfs18 opened 1 year ago

bfs18 commented 1 year ago

The VAE code does not contain a sample function. Should I sample z_k from N(0, 1) and then decode it to get a sample from the model?