search

wilsonmr / anvil

Repository containing code related to flow based generative model
https://wilsonmr.github.io/anvil/
GNU General Public License v3.0
0 stars 2 forks source link

New coupling layer: "non-compact projection" #47

Closed jmarshrossney closed 3 years ago

jmarshrossney commented 4 years ago

An alternative to wrapping Real NVP in projection and inverse projection maps. Instead, each coupling layer is a composition of a projection, a single affine transformation, and an inverse projection.

This is implemented as described in section 2.1.3 of https://arxiv.org/pdf/2002.02428.pdf. However, I haven't linearised the transformation close to 0 and 2pi, since so far all attempts have been computationally expensive and haven't noticeable improved the training.

jmarshrossney commented 3 years ago

Not required for phi four