Skip to content

FR: IWAE_DReG importance weighted VAE with doubly reparametrized gradients #2598

@fritzo

Description

@fritzo

This proposes adding an IWAE_DReG loss similar to the ELBO class but implementing the doubly reparametrized gradient estimator. @iffsid points out pytorch/pytorch#25783 that this would be much easier if Distribution objects had a .detach() method.

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions