-
Notifications
You must be signed in to change notification settings - Fork 2
Factorized log probability method, general variational bound class #10
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
base: main
Are you sure you want to change the base?
Conversation
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
suhasshrinivasan
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I decided to rollback ELU implementation, since changing its constructor broke backwards compatibility. We would need to discuss the modification once again.
|
In |
We want the log probabilities of a random variable dimension by dimension when the distribution is factorized for a few cases such as density inspection. This PR adds factorized log probability computation methods in the core classes such as IndependentTrainableDistributionAdapter and also extends the idea to factorized log dets in transforms in normalizing flows and factorized elbo and iw bound in variational dequantized distributions. The functionality is lightly tested and elucidated in a jupyter notebook in
notebooks/factorized_log_prob.ipynb.