Hi,
any pointers on how to implement semi-amortized VAE (https://arxiv.org/pdf/1802.02550.pdf) in pyro?
thanks
Hi,
any pointers on how to implement semi-amortized VAE (https://arxiv.org/pdf/1802.02550.pdf) in pyro?
thanks
If you use a pyro.optim.MultiOptimizer
you can differentiate through parameter updates as described in the paper; the rest of the algorithm seems like pretty straightforward application of existing Pyro machinery. See the custom SVI objectives tutorial for more details.
If you can ask a more specific question with code we can probably be more helpful.