Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/ncp circle #3

Merged
merged 5 commits into from
Jun 16, 2020
Merged

Feature/ncp circle #3

merged 5 commits into from
Jun 16, 2020

Conversation

jmarshrossney
Copy link
Owner

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.

marshrossney added 5 commits June 11, 2020 09:44
correct single line
circle projection with nn.Parameters

fix ncp circle

add docstring and example runcard

upgrade ncp to coupling layer

correct single line

add option to specify final activation
@jmarshrossney jmarshrossney merged commit 577d1b5 into master Jun 16, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant