Variational Autoencoders — Pyro Tutorials 1.8.0 documentation
pyro.ai/examples/vae.htmlSo, for example, when we call parameters () on an instance of VAE, PyTorch will know to return all the relevant parameters. It also means that if we’re running on a GPU, the call to cuda () will move all the parameters of all the (sub)modules into GPU memory. Inference ¶ We’re now ready for inference. Refer to the full code in the next section.