-
Notifications
You must be signed in to change notification settings - Fork 5
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
more NF examples #11
base: main
Are you sure you want to change the base?
more NF examples #11
Conversation
This reverts commit 1c1c88a.
@zuhengxu, can you push to merge this PR if nothing significant is standing? Thanks! |
No prob! I'll get back to this PR this weekend and hopefully merge it ASAP. |
Could we add a script to test whether all the example models run, e.g. on |
Yes, let me clean up the Ham VI demo and then add the test script. Currently I can confirm that all the examples run on CPUs by doing
|
@yebai Currently all the examples are only tested on CPUs. Do you have examples of such script (for testing runs) that I can refer to? (I did do In terms of GPU, shall we first merge this PR and do GPU examples/test (#49) on a separate PR? the purpose this PR is just to demonstrate how to define and customize flow layers (at least for those popular flows). I think that to properly take advantage of GPU, we might want some interface adjustment. |
as per issue #15
NormalizingFlows.jl
+Bijectors.jl
(a tentative list)