Implementations of RealNVP in Pytorch/Pyro.
In this repository we implement Normalizing Flows for both Unconditional Density Estimation (i.e., ) and Conditional Density Estimation (i.e., ). The repository is organized as follows:
- models: contains .py scripts of Unconditional and Conditional AffineCoupling Flows respectively
- (Un)conditionalNF.ipynb: basic usage example on 2-d density estimation
- cnf_torch_save_run: trained model for the Conditional Normalizing Flow notebook for faster replication