Semi-amortized VAE


#1

Hi,

any pointers on how to implement semi-amortized VAE (https://arxiv.org/pdf/1802.02550.pdf) in pyro?

thanks


#2

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.